作者回复: 你好,同学 感谢你的认真阅读和提问 这里我将如何得到best.pt做一个更加细节的回复,希望能解答你的疑惑 在 YOLOv5 的训练过程中,best.pt是系统自动保存的“最佳模型”权重文件,表示在验证集上表现最好的模型状态。 它的生成过程如下: 首先是训练过程中的评估模型 在设置超参数时候,我们设定了epoch值,一个epoch代表一个训练周期,每个训练周期(epoch)结束后,模型会在验证集上计算性能指标,比如 精确度(Precision),召回率(Recall)和 平均精度(mAP) 然后选择最佳模型 YOLOv5使用一个加权组合的指标fitness来确定模型的性能,这个指标是精确度、召回率、mAP@0.5和mAP@0.5-0.95的加权和。 fitness计算公式为fi = 0.1*mAP@.5 + 0.9*mAP@.5-.95,这个计算的变化趋势也可以在训练结果的可视化图片上看到 最后就是更新和保存最佳模型 如果当前epoch的fitness值高于之前保存的最佳模型的fitness值,那么当前模型将被认为是新的“最佳模型”,并保存为best.pt文件。通常保存在runs/train/exp/weights/目录下,其中exp是实验的名称或编号。 总结来说,best.pt是在YOLOv5训练过程中,根据验证集上的性能指标自动选择并保存的最佳模型权重文件。 加油~