yolov7模型训练结果分析以及如何评估yolov7模型训练的效果

发布于:2022-12-06 ⋅ 阅读:(6220) ⋅ 点赞:(6)

最近是刚刚训练yolov7模型,但是只会一股脑的训练是不行的,要懂得训练多少epoch,以及通过哪些指标来查看训练的效果如何,现在这几天的经验总结一下。

本实验以person为例子,分别训练100epoch、60epoch、50epoch

训练经验:由于并不知道到底训练多少epoch效果比较好,所以现在先设置成100。(这几天查询资料得出来的结论一般50多就差不多了)

1、关于yolov7训练结果的文件解析

一、weights

也就是训练好的模型,一般使用best.pt去进行推理

 包括best.pt和last.pt,以及默认25epoch保存一次模型,以及保存最后5个epoch的模型

二、 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的所有结果中,预测正确的比重

 三、F1_curve.png

F1分数,它被定义为查准率和召回率的调和平均数

一些多分类问题的机器学习竞赛,常常将F1-score作为最终测评的方法。它是精确率和召回率的调和平均数,最大为1,最小为0。

F1-Score的值是从0到1的,1是最好,0是最差。

 这是100epoch得到的F1_curve,说明在置信度为0.4-0.6区间内得到比较好的F1分数

四、hyp.yaml和opt.yaml

训练时的超参数以及train.py中间的参数

五、P_curve.png

准确率precision和置信度confidence的关系图

 六、PR_curve.png

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

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

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


七、R_curve.png

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

 八、results.png

Box:Box推测为GIoU损失函数均值,越小方框越准;
Objectness:推测为目标检测loss均值,越小目标检测越准;
Classification:推测为分类loss均值,越小分类越准,本实验为一类所以为0;
Precision:精度(找对的正类/所有找到的正类);

Recall:真实为positive的准确率,即正样本有多少被找出来了(召回了多少)。

Recall从真实结果角度出发,描述了测试集中的真实正例有多少被二分类器挑选了出来,即真实的正例有多少被该二分类器召回。

val BOX:  验证集bounding box损失

val Objectness:验证集目标检测loss均值

val classification:验证集分类loss均值,本实验为一类所以为0

mAP是用Precision和Recall作为两轴作图后围成的面积,m表示平均,@后面的数表示判定iou为正负样本的阈值,@0.5:0.95表示阈值取0.5:0.05:0.95后取均值。

mAP@.5:.95(mAP@[.5:.95])
表示在不同IoU阈值(从0.5到0.95,步长0.05)(0.5、0.55、0.6、0.65、0.7、0.75、0.8、0.85、0.9、0.95)上的平均mAP。

mAP@.5:表示阈值大于0.5的平均mAP

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

九、results.txt

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

十、train_batchx

我设置的batch_size为8所以一次读取8张图片

 十一、test_batchx_labels

验证集第一轮的实际标签

十二、运行tensorboard

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

二、分析模型的训练效果

接下来我将分别展示100epoch、60epoch和50epoch的tensorboard的可视化训练结果

100epoch

Loss

AP值

60epoch

Loss

AP值

50epoch

Loss

AP值

 分析:从100epoch的Loss来看从50-60epoch损失值就维持在一定的范围内,所以我们可以缩小训练epoch,然后设置epoch为60继续训练,发现相较于100epoch的AP值86.8,60epoch的AP值为87.7上升0.9,但是发现val的目标检测损失在50epoch之后有所上涨,所以我们再次设置epoch为50进行训练,发现AP值上升0.1。

总结:所以来说查看yolov7模型训练的效果要综合四个损失函数的值来看,先设置的epoch次数多一些,然后通过损失函数来看减少或增加训练轮数,然后查看AP值是否有所增加。

最后增加一下计算模型训练的FPS值,设置相应的参数并运行test.py

运行结果

 FPS=1000/(4.3+0.6+5.0)=101

本文含有隐藏内容,请 开通VIP 后查看

深圳SEO优化公司龙岗网站改版推荐商洛百搜标王木棉湾设计网站推荐渭南百姓网标王推广哪家好西乡百姓网标王推广推荐盘锦百搜标王霍邱网站优化推广价格台州seo网站优化推荐安康建设网站公司新乡关键词按天扣费多少钱兰州外贸网站设计价格永新阿里店铺运营多少钱银川网站开发价格布吉企业网站设计报价大庆网络广告推广推荐白山高端网站设计公司萍乡设计公司网站多少钱临猗网站制作推荐宣城网站优化按天扣费报价罗湖seo网站推广忻州网站改版价格南山模板推广报价榆林网站制作设计多少钱金华外贸网站制作推荐宿迁优化承德百度关键词包年推广公司济南网站定制价格惠州网站制作报价邵阳SEO按效果付费推荐开封至尊标王歼20紧急升空逼退外机英媒称团队夜以继日筹划王妃复出草木蔓发 春山在望成都发生巨响 当地回应60岁老人炒菠菜未焯水致肾病恶化男子涉嫌走私被判11年却一天牢没坐劳斯莱斯右转逼停直行车网传落水者说“没让你救”系谣言广东通报13岁男孩性侵女童不予立案贵州小伙回应在美国卖三蹦子火了淀粉肠小王子日销售额涨超10倍有个姐真把千机伞做出来了近3万元金手镯仅含足金十克呼北高速交通事故已致14人死亡杨洋拄拐现身医院国产伟哥去年销售近13亿男子给前妻转账 现任妻子起诉要回新基金只募集到26元还是员工自购男孩疑遭霸凌 家长讨说法被踢出群充个话费竟沦为间接洗钱工具新的一天从800个哈欠开始单亲妈妈陷入热恋 14岁儿子报警#春分立蛋大挑战#中国投资客涌入日本东京买房两大学生合买彩票中奖一人不认账新加坡主帅:唯一目标击败中国队月嫂回应掌掴婴儿是在赶虫子19岁小伙救下5人后溺亡 多方发声清明节放假3天调休1天张家界的山上“长”满了韩国人?开封王婆为何火了主播靠辱骂母亲走红被批捕封号代拍被何赛飞拿着魔杖追着打阿根廷将发行1万与2万面值的纸币库克现身上海为江西彩礼“减负”的“试婚人”因自嘲式简历走红的教授更新简介殡仪馆花卉高于市场价3倍还重复用网友称在豆瓣酱里吃出老鼠头315晚会后胖东来又人满为患了网友建议重庆地铁不准乘客携带菜筐特朗普谈“凯特王妃P图照”罗斯否认插足凯特王妃婚姻青海通报栏杆断裂小学生跌落住进ICU恒大被罚41.75亿到底怎么缴湖南一县政协主席疑涉刑案被控制茶百道就改标签日期致歉王树国3次鞠躬告别西交大师生张立群任西安交通大学校长杨倩无缘巴黎奥运

深圳SEO优化公司 XML地图 TXT地图 虚拟主机 SEO 网站制作 网站优化