【Week Y6】修改yolov5s的网络结构

• 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
• 🍖 原作者：K同学啊 | 接辅导、项目定制

修改要求：
(1)原第4层的C3*2修改为C2*2
(2)原第6层的C3*3修改为C3*1
(3)原第7、8层删除（注意删去这两层后的层索引值的变化）

原YOLOv5s结构：

修改后的结构：

一、修改前的yolov5s.yaml

二、修改后的yolov5s.yaml

注意仔细观察网络结构的内容，以及每一层的索引号，都是有关联的，修改时要结合网络结构图，一一对应。

还需要注意head的第一个卷积层的输入size，要和前面层数对齐。

三、训练

执行：python .\train.py --img 900 --batch 2 --epoch 100 --data .\fruit.yaml --cfg .\models\yolov5s.yaml --weights .\yolov5s.pt --device cpu

得到：

训练正常：