发布时间:2023-04-20 文章分类:电脑百科 投稿人:赵颖 字号: 默认 | | 超大 打印

yolov5-runs文件中对train结果的说明

1、weights

训练所得权重

2、confusion_matrix

混淆矩阵
列代表预测的类别,行代表实际的类别。其对角线上的值表示预测正确的数量比例,非对角线元素则是预测错误的部分。混淆矩阵的对角线值越高越好,这表明许多预测是正确的。

3、F1_curve

F1得分与置信度关系
x轴为置信度,y轴为F1得分

4、hyp.yaml

训练相关超参数

5、labels

一共四张图
左一:每个类别的数据量
右一:labels的 bounding_box
左二:labels的中心点坐标
右二:labels的矩阵宽高

6、labels_correlogram

labels的中心点x,y和矩阵宽高w,h
顶端对角线上:各自的分布直方图
其余位置:相互之间的分布情况

7、opt.yaml

最优参数

8、P_curve

准确率与置信度关系

9、PR_curve

PR曲线中的P代表的是precision(精准率),R代表的是recall(召回率),其代表的是精准率与召回率的关系,一般情况下,将recall设置为横坐标,precision设置为纵坐标。PR曲线下围成的面积即AP,所有类别AP平均值即Map.

如果PR图的其中的一个曲线A完全包住另一个学习器的曲线B,则可断言A的性能优于B,当A和B发生交叉时,可以根据曲线下方的面积大小来进行比较。一般训练结果主要观察精度和召回率波动情况(波动不是很大则训练效果较好)

Precision和Recall往往是一对矛盾的性能度量指标;
提高Precision == 提高二分类器预测正例门槛 == 使得二分类器预测的正例尽可能是真实正例;
提高Recall == 降低二分类器预测正例门槛 == 使得二分类器尽可能将真实的正例挑选

10、R_curve

召回率和置信度之间的关系

11、results.csv

每一次迭代对应的

12、results

部分参考:
https://blog.csdn.net/weixin_44570845/article/details/121337026
https://github.com/ultralytics/yolov5/issues/5138