如何使用val.py在YOLOv5的自定义测试数据集上计算mAP？

Question

我正在尝试使用以下方法计算我的测试数据集上的 mAP：

!python val.py  --task test

我正在使用自定义

data.yaml

文件。但我不知道如何指出这一点

data.yaml

。当我使用以下代码时：

val.py --data= /content/data.yaml --task test

模型输出

invalid argument

错误。默认情况下，它也会尝试计算

coco.yaml

上的 mAP。

Answer 1

您应该使用 val 而不是任务参数的测试值，如下所示。

python val.py --weights {{model}} --data data.yaml --task val

Answer 2

使用你的方法，我成功地使用 val.py 计算了测试集的 mAP。它打印了图像和数值结果。我使用的命令如下所示：

python val.py --weights /content/last.pt --data /content/data.yaml --task test

我希望它有帮助，但似乎已经过去一年了。