ValueError：发现样本数量不一致的输入变量：[6, 1]

Question

我正在突破图像分类按钮。 我想计算每个图像的精度和召回率。 目前，对于每张图像，我有两个 .txt：一个表示 GT，一个表示检测。 例子：

GT: [59, 59, 59]
DETECTION: [30, 30, 1]

如果 GT 的长度小于或大于检测的长度，如何计算指标？

GT: [64, 64, 64, 64, 64]
DETECTION: [34]

或者

GT: [59, 59, 59]
PRED: [30, 30, 1, 59]

我尝试使用此函数： precision = precision_score(GT, DETECTION, average = 'macro' ) 但是当len(GT)与len(DETECTION)不同时，它无法计算指标...

ValueError: Found input variables with inconsistent numbers of samples: [6, 1]

有谁知道如何帮助我？ 提前致谢

Answer 1

不，您必须使用在两种情况下都具有相同形状的数组。 例如，如果您有 K 个类，那么两个向量都应该具有形状 (1, K)。 然后值是 0 或 1，具体取决于类是否存在。 您必须对类别进行编码。