一、前言
文件位置:
**./models/yolo.py**
这个文件是YOLOv5网络模型的搭建文件,如果你想改进YOLOv5,那么这么文件是你必须进行修改的文件之一。文件内容看起来多,其实真正有用的代码不多的,重点理解好我文中提到的一个函数两个类即可。
注:由于YOLOv5版本众多,同一个文件对于细节处你可能会看到不同的版本,不用担心这都是正常的,注意把握好整体架构即可。
- 📌 本周任务:将YOLOv5s网络模型中的C3模块按照下图方式修改形成C2模块,并将C2模块插入第2层与第3层之间,且跑通YOLOv5s。
- 💫 任务提示:
- 提示1:需要修改
common.yaml
、yolo.py
、yolov5s.yaml
文件。 - 提示2:
C2
模块与C3
模块是非常相似的两个模块,我们要插入C2
到模型当中,只需要找到哪里有C3
模块,然后在其附近加上C2
即可。
- 提示1:需要修改