python - 将 sklearn.metrics Jaccard Index 与图像一起使用？

Question

我正在尝试进行一些图像比较，首先是找到 Jaccard 指数。我正在使用 Jaccard Index 的 sklearn.metrics 实现 使用下面的示例和一小部分数字，它按预期工作。

import numpy as np
from sklearn.metrics import jaccard_similarity_score

#The y_pred represents the values that the program has found 
y_pred = [0,0,1,0,0,0,1,1,1,1,0,1,0,1,0,0,1,0,1,1,1,0,1,1,0,1,1,1,1,1,0,1,0,1,1,1,0,0,0,0,1,1,0,0,1,1,0,1,1,1]
#The y_true represents the values that are actually correct 
y_true = [1,0,0,1,0,1,1,0,1,1,1,0,1,0,1,1,0,1,1,0,0,1,0,1,0,1,0,1,0,1,1,1,1,1,0,1,1,0,0,0,0,1,1,1,0,1,0,1,1,1]

iou = jaccard_similarity_score(y_true, y_pred)

虽然它给出了...的错误

ValueError: unknown is not supported

当我给它输入两张图片时，比如......

iou = jaccard_similarity_score(img_true, img_pred)

我不确定该怎么做，我尝试使用 OpenCV 将图像转换为灰度，并将两个图像都设为 astype(float)，但在这两种情况下都不走运。

Answer 1

作为答案发布，这样问题就可以结束了:扁平化 img_true 和 img_pred 通过 img_true.flatten() 和 img_pred 解决.flatten()