替换YOLOv5主干网络_yolov5主干网络替换

作者：你好赵伟 | 2024-03-01 04:25:29

踩

yolov5主干网络替换

首先写一个 demo.py 文件来查看 torchvision下面都有哪些模型

如图所示，可以看到 torchvision.models 下面有很多模型, 比如 vgg16,resnet50,AlexNet,resnet18等等，在这里我选择 vgg16 模型来替换YOLOv5的主干网络

下面先来看一下 vgg16 模型的结构：

可以看到 vgg16 模型主要由 feature (特征提取层：做特征提取) 、avgpool (池化层) 以及 classifier (分类层) 组成，要替换YOLOv5的主干网络，其实就是要将YOLOv5的特征提取层用vgg16的特征提取层进行替换

下面给该网络模型一个输入数据，来看一下该网络模型每一层的输出大小是多少，才能去进行主干网络的替换操作

打开 yolov5s.yaml 文件，如图所示，输出要有 80x80,40x40

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/170965