【发布时间】:2021-09-28 14:53:09
【问题描述】:
想问下深度学习中的分割任务除了IOU(intersection over union)还有其他好的指标吗?
因为有时我从 IOU 得到 NaN 结果,只是想知道是否还有一些其他指标可以帮助观察模型的性能。
def IoU(
targets: np.array,
outputs: np.array,
) -> np.array:
intersection = np.sum(
outputs * targets,
axis=(0, 1)
)
union = np.sum(
outputs + targets,
axis=(0, 1)
) - intersection
return np.mean(intersection / union)
【问题讨论】:
-
问题是关于方法而不是代码。请看
help center。此外,您的 IOU 实现中可能存在错误。 -
@AbhishekPrajapat 对不起,我只是重写它
标签: image-processing deep-learning computer-vision image-segmentation semantic-segmentation