yolov5训练结果解析

yolov5训练结果的文件解析

1、weights训练好的模型,一般使用best.pt去进行推理
2、confusion_matrix.png
请添加图片描述
混淆矩阵以矩阵形式将数据集中的记录按照真实的类别与分类模型预测的类别判断两个标准进行汇总。其中矩阵的行表示真实值,矩阵的列表示预测值。

TP(True Positive): 将正类预测为正类数 即正确预测,真实为0,预测也为0

FN (False Negative):将正类预测为负类 即错误预测,真实为0,预测为1

FP(False Positive):将负类预测为正类数 即错误预测, 真实为1,预测为0

TN (True Negative):将负类预测为负类数,即正确预测,真实为1,预测也为1

精确率和召回率:

精确率Precision=TP / (TP+FP), 在预测是Positive所有结果中,预测正确的比重
召回率recall=TP / (TP+FN), 在真实值为Positive的所有结果中,预测正确的比重

3、F1_curve.png
精确率和召回率的调和平均数,最大为1,最小为0。
F1-Score的值是从0到1的,1是最好,0是最差。


说明在置信度为0.4-0.6区间内得到比较好的F1分数
4、hyp.yaml和opt.yaml
训练时的超参数以及train.py中间的参数
5、P_curve.png
准确率precision和置信度confidence的关系图

6、PR_curve.png

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

一般训练结果主要观察精度和召回率波动情况(波动不是很大则训练效果较好)
观察mAP@0.5 & mAP@0.5:0.95 评价训练结果。

9、results.txt
训练次数、GPU消耗、训练集边界框损失、训练集目标检测损失、训练集分类损失、训练集总损失、targets目标、输入图片大小、Precision、Recall、mAP@.5、mAP@.5:.95、验证集边界框损失、验证集目标检测损失、验证机分类损失

10、train_batchx
我设置的batch_size为16,所以一次读,16张图片
11、test_batchx_labels
验证集第一轮的实际标签
12、运行tensorboard

activate yolov7(自己所配的环境名称)
tensorboard --logdir=训练结果所在的文件夹

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2023年6月4日
下一篇 2023年6月5日

相关推荐