yolov5中的best.pt是如何确定的

在yolov5 的使用过程中几乎都会发现的问题:

训练结果有last.ptbest.pt ,

last.pt好理解,就是最后一个epoch的输出,但是best是啥意思?怎么才算best?

我们来一行行看train.py源码

yolov5中的best.pt是如何确定的_第1张图片

追溯到./utils/metrics.py中的fitness函数,可以看到是将[email protected][email protected]:0.95按照1:9的比重确定最佳模型的

yolov5中的best.pt是如何确定的_第2张图片

你可能感兴趣的:(YOLOv5解析,YOLO)