插入C2模块到指定位置
- 一、common.py文件修改
- 二、yolo.py文件修改
- 三、yolov5s.yaml修改
- 四、训练
- 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
- 🍖 原作者：K同学啊 | 接辅导、项目定制
模块结构如下：【同Y4】
【Y4】周直接在C3的模块中减少了一次Conv，本次需要还原C3模块，并重新定义一个C2模块。
C2模块插入位置如下：
一、common.py文件修改
在原C3模块附近，增加C2模块定义：
二、yolo.py文件修改
找到parse_model()模块，按如下修改：
parse_model()模块用于将模型的各个模块拼接起来，从而形成一个完整的网络结构。
三、yolov5s.yaml修改
找到backbone位置，按如下修改：
四、训练
执行：python .\train.py --img 900 --batch 2 --epoch 100 --data .\fruit.yaml --cfg .\models\yolov5s.yaml --weights .\yolov5s.pt --device cpu
打印的网络结构如下：
训练过程正常：
