yolo - 如何测试 yolov3 和 yolov4 权重文件？

Question

我正在尝试使用 yolov3 和 yolov4 进行对象检测，我只想训练 1 类 = 人。我正在使用 COCO 数据集。

训练后如何测试我的体重文件？我想测试所有“测试数据集”，而不仅仅是一张图像。

我有火车损失图，但这还不够。我想计算准确率、精确度、召回率等。我发现了这个例子：https ://darknet.gong.im/

./darknet 检测器测试 cfg/coco.data cfg/yolov3.cfg weights/yolov3.weights batch ./in_images/ ./out_images/ >./results.txt

您能告诉我如何将这两种算法与成功指标进行比较吗？

如何测试 yolov3 和 yolov4 权重文件？

Answer 1

有用于测试模型的地图功能。

./darknet 检测器图 cfg/coco.data cfg/yolov3.cfg weights/yolov3.weights

如果您标记测试数据集并将其路径提供给数据文件中的“有效”字段，则可以在数据集上使用 map 函数。

您还可以在训练时使用 -map 标志来查看图表上的地图结果。如果您在训练暗网时使用 -map 标志，也会保存最佳权重文件，因此您无需找到最佳权重文件。