【问题标题】:How to get MAP score of the Yolov4 Model?如何获得 Yolov4 模型的 MAP 分数?
【发布时间】:2021-09-16 18:13:09
【问题描述】:

我有一组图像和 Yolo 注释文件(txt 格式)用于验证。

如何使用 darknet 框架(Repo:https://github.com/AlexeyAB/darknet)正确使用 -map 参数(如下所示)来获取验证数据集的 mAP 分数?

/darknet detector map cfg/test.data cfg/test_tiny.cfg backup\my_yolov3_tiny_final.weights 

是否有可能从中推导出混淆矩阵?

【问题讨论】:

  • 请提供更多信息,例如您使用哪个repo进行培训,如果您自己编写代码,请分享。
  • @CuCaRot 感谢您指出这一点。我已经添加了回购细节。只是想知道是否有任何现有方法可以获取验证数据集上的 mAP 分数。谢谢!
  • that repo how to train yolov4,查看第 8.1 节
  • 我明白你的意思了,在这种情况下,使用预训练模型来预测验证集,然后你可以使用this repo 来评估它。这是我最喜欢的 repo,因为它很简单。
  • 感谢您的建议!这就是我一直在寻找的。我会检查的。

标签: machine-learning darknet yolov4


【解决方案1】:

/darknetdetector mapdetector map cfg/test.data cfg/test_tiny.cfg backup\my_yolov3_tiny_final.weights 其中 test.data 是您正在训练的同一个数据文件。您可以在训练时使用 -map 标志,因此在 4 个 epoch(4 * train_size / batch)之后,您将看到 mAP@0.5。

【讨论】:

    【解决方案2】:

    我无法编辑上面的代码。如需更正确的命令:

    ./darknet detector map test.data yolov4-tiny.cfg yolov4-tiny_last.weights

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2020-12-26
      • 1970-01-01
      • 1970-01-01
      • 2021-09-11
      • 1970-01-01
      • 2018-11-03
      • 2021-05-10
      • 1970-01-01
      相关资源
      最近更新 更多