1、前言
本章将介绍如何将官方的定义的vgg16代码更换到yolov5的主干网络
在之前实现了vgg的更换代码,只不过上次的vgg是自己定义在py脚本中的,代码进行稍微更改才行。
这里官方实现的更为简单,而且相同网络结构的话,可以跟着本章自己更改
具体的还是参考这一章:YOLOV5 改进:替换backbone(MobileNet为例)-CSDN博客
Tips :之前在这篇博文中YOLOV5 改进:替换backbone为Vgg_yolov5替换vgg-CSDN博客
有个兄弟问可不可以将训练好的再次进行迁移学习,当然是可以的。
因为迁移学习只是将网络参数进行载入,其实就是参数初始化,既然网络是一样的,当然可以载入,和其他的没有关系。
至于yolov5s.pt 为什么每次都可以载入我们自定义的数据集,因为官方的代码对载入权重进行了判断,只会把相同的部分载入,不相同的丢弃即可,其实就是很简单的迁移学习策略
为了方便理解,本章将训练200个epoch,然后载入训