【YOLOv7训练】——预训练重使用_yolov7预训练权重_yolov7预训练的模型能用吗

文章目录

• 更新提醒：2023/04/01更新
• YOLOv7简介
• 预训练权重（用不用？用哪个？）
• 总结

更新提醒：2023/04/01更新

YOLOv7简介

YOLOv7论文链接：YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
官方github代码链接：https://github.com/WongKinYiu/yolov7

YOLOv7于2022.07发布，已被CVPR2023接收！

此贴记录自己使用YOLOv7训练自己数据集时权重使用问题

最后个人建议，能不用YOLOv7就别用！！！别用！！！别用！！！
训练真的太慢了！！！训练真的太慢了！！！训练真的太慢了！！！（OTA的锅，关了OTA会快好几倍，但性能会下降）
太吃显存了！！！太吃显存了！！！太吃显存了！！！
收敛曲线很魔鬼！！！收敛曲线很魔鬼！！！收敛曲线很魔鬼！！！
重要的事情说三遍！！！重要的事情说三遍！！！重要的事情说三遍！！！

我知道说三遍，你还是不会听的！！！

那就开始正文了，不罗嗦了！

预训练权重（用不用？用哪个？）

一般训练，我都会优先使用官方提供的权重进行训练，而且v7和v5代码架构相同，同样可以自动判断继承权重。（具体什么意思呢？简单概括就是，加载预训练权重时，会先对你真正使用主干网络跟预训练权重包含的主干网络参数进行比较，观察有多少层是相同的，然后就只会加载相同的层数训练。所以，后面即使更改了主干结构，依旧可以使用预训练权重。PS：如果存在以为，自己可以测试，亦或者不用预训练权重。）

预训练权重的选择
yolov7官方提供了很多种权重，而且有的结构提供了多种结构？比如：yolov7.pt和yolov7_training.pt

这两个权重到底用哪个？什么区别？
使用时，会发现两个权重都可以训练，而且都有预训练的效果，结果好像差不多，那设置两个权重的意义是？
yolov7_traing.pt结构是未重参数化的结构；而yolov7.pt是合并分支后的网络结构，也就是重参数化后的模型结构，为什么也可以训练？
yolov7_traing.pt训练的时候用的是training中的yolov7.yaml，yolov7.pt训练的时候用的是deploy中的yolov7.yaml

作者在代码里的解释：

总结

说了一堆，总结就是：
1、其实，不用权重一点关系没有！亲测，我训练时，用权重和不用权重数据一模一样（真奇了怪了，可能我都是300epoch训练，后面自己就收敛了，预训练权重也就是收敛快些，也不会提点吧好像）
2、用了权重，可能会掉点，还不如不用（这是可能的，原因我也不知道）
3、如果非要用，那训练时就用yolov7_training.pt作预训练权重（其实我建议要是时间够，两个都试试，哪个好用哪个呗。没时间的话，就一直挑一个用，也是行的，控制变量法）

本篇博文的感觉就是，说了等于没说！看官莫怪！

现在能在网上找到很多很多的学习资源，有免费的也有收费的，当我拿到1套比较全的学习资源之前，我并没着急去看第1节，我而是去审视这套资源是否值得学习，有时候也会去问一些学长的意见，如果可以之后，我会对这套学习资源做1个学习计划，我的学习计划主要包括规划图和学习进度表。

分享给大家这份我薅到的免费视频资料，质量还不错，大家可以跟着学习

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友，可以戳这里无偿获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！