trainer类是如何选择YOLOv8最佳损失模型的?

Aus*_*tin 3 python loss yolo

从 YOLOv8 文档中,我不清楚 YOLOv8 训练器类使用哪个损失指标来确定训练运行中保存的最佳损失模型。是基于验证还是训练损失?

具体来说,当我查看 YOLOv8 训练运行的输出时,我没有看到任何元数据表明哪个纪元导致保存在 running/train/weights/best.pt 的最佳损失模型

Mik*_*e B 6

save_model函数中你可以看到它使用最大值fitness来保存最好的模型。适应被定义为 4 个指标的加权组合[P, R, mAP@0.5, mAP@0.5:0.95]PR因某种原因而被忽视。mAP@0.5mAP@0.5:0.95分别加权0.1和0.9。

如果fitness找不到,则使用loss代替