什么是 yolo_best.weights 文件？它与 yolo_custom.weights 有何不同

Posted 2023-03-16

【中文标题】什么是 yolo_best.weights 文件？它与 yolo_custom.weights 有何不同

【英文标题】：what is yolo_best.weights file? How is it different from yolo_custom.weights

【发布时间】：2020-10-14 14:36:27

【问题描述】：

我正在 Yolo v4 上训练一个自定义模型以进行对象检测，我知道 Yolo 每 100 次迭代都会保存“yolo_custom.weights”文件和“yolo_custom_XXXX.weights " 每 1000 次迭代。

除此之外，我还有一个名为“yolo_custom_best.weights”文件的新文件，它会像“yolo_custom.weights”文件一样不断更新一些迭代次数.

我正在使用暗网使用以下命令训练我的模型： ./darknet detector train "data/obj.data" "cfg/yolov4_custom.cfg" yolov4.conv.137 -dont_show -map

我在训练时第一次使用-map 标志。

我的问题是：

什么是“yolo_custom_best.weights”文件？它是因为我使用-map 标志而形成的吗？使用该文件检测是否比“yolo_custom.weights”文件更好？ 如果我想从某个点恢复训练（如果我的 Google Colab 运行时结束），我应该使用哪个文件来恢复？ custom 或 custom_best? 谁能解释一下什么是mAP？是否会导致此文件生成和更新？

谢谢！

【参考方案1】：

“最佳权重”文件在使用 -map 标志时形成。 当您的 Colab 运行时断开连接时，我会说最好从 yolo_last.weights 文件恢复训练。这可以通过以下方式完成：

!./darknet detector train data/obj.data cfg/yolov4-obj.cfg yolov4-obj_last.weights -dont_show

这将从上次更新的位置获取训练。

mAP（平均平均精度）是所有类别的 AP（平均精度）的平均值（我在这里可能错了）。 Here is a useful link to explain mAP。