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


修改后的 yaml 文件用到了 ELAN、ELAN_H、MP_1 和 MP_2 四个模块



















