本博文将记录和分享个人的一些学习工作体会和踩过的坑,希望对大家能有帮助。
一、对比实验
用yolov5l.ymal和yolov5n.ymal文件,然后加载yolov5s.pt后,训练得到的模型效果极差(mAP差30%),重新训练后换为yolov5l.pt就好了。证明预训练模型还是需要匹配上的。对实验影响较大,也可能设置的轮次不够才100轮,没有进行收敛还。
二、TensorBoard
如果有朋友是租远程服务器AutoDL-品质GPU租用平台-租GPU就上AutoDL
想要在线监控训练情况的话,可以将train中的项目名称改为这个目录。
然后打开网页就可以实时监控训练情况,当mAP以及Loss稳定下来后,就意味着模型进行了收敛,即可停止进行训练了。同时也可以对比改进方法与baseline的提升效果等,相当方便了。
三、YOLO算法输出
有朋友问,yolo算法head输出部分的30是如何得到的,公式=(5(四个坐标+类别名称)+分类数50)*3 (3是三个anchor)=30得出的。