文章目录
-
- 修改方式
-
- common.py
- yolo.py
- YOLOv7-ELAN.yaml
原始的 YOLOv7
yaml
文件的模块是拆开写的,比较乱,
改进起来也不太容易,这篇博文将 YOLOv7
yaml
文件换了一种写法,
参数量和计算量是完全和原来一致的,区别只是在于 yaml
文件的写法不同,
封装后具体的结构可以参考我下面的这个结构图,图片的每个模块和 yaml
中的模块是一一对应的,
修改后的 yaml
文件用到了 ELAN
、ELAN_H
、MP_1
和 MP_2
四个模块