使用sklearn.metrics Jaccard Index和图像？

Question

我正在尝试进行一些图像比较，首先找到Jaccard Index。我使用了Jaccard Index的sklearn.metrics实现使用下面的例子只有一小部分数字，它就像预期的那样工作。

ValueError: unknown is not supported

虽然它给出了错误...

iou = jaccard_similarity_score(img_true, img_pred)

当我给它喂食两个图像，如....

{{1}}

我不确定该怎么做，我尝试使用OpenCV将图像转换为灰度图像，并使这两种图像都显示为astype（float），但两种情况都没有运气。

Answer 1

您可以使用ravel()将其转换为1-D：

img_true=np.array(img_true).ravel()
img_pred=np.array(img_pred).ravel()
iou = jaccard_similarity_score(img_true, img_pred)

Answer 2

发布回答以便关闭问题：通过执行img_true和img_pred

解决展平img_true.flatten()和img_pred.flatten()