YOLOv8最佳损失模型是如何被trainer类选出来的？

Question

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

具体来说，当我查看 YOLOv8 训练运行的输出时，我没有看到任何元数据表明哪个时期产生了保存在 runs/train/weights/best.pt 中的最佳损失模型

Answer 1

在save_model函数中可以看到它使用最大

fitness

来保存最好的模型。 fitness 被定义为 4 个指标

[P, R, [email protected], [email protected]:0.95]

的加权组合。

P

和

R

由于某种原因被忽略。

[email protected]

,

[email protected]:0.95

的权重分别为 0.1 和 0.9.

如果找不到

fitness

，则用loss代替