【Week Y5】yolo.py文件解读，插入C2模块到指定位置

本文主要是介绍【Week Y5】yolo.py文件解读，插入C2模块到指定位置，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

• 🍨 本文为🔗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

打印的网络结构如下：

训练过程正常：

这篇关于【Week Y5】yolo.py文件解读，插入C2模块到指定位置的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---