文章目录
1、自己动手制造一个YOLOv9-S网络结构
1.1 改前改后的网络结构(参数量、计算量)对比
改前的yolov9.yaml
参数量58.35M,计算量267.1GFLOPs
改后的yolov9-S.yaml
参数量15.60M,计算量67.7GFLOPs
- 这是修改
前
调用的yolo.py
测试的yolov9.yaml
的打印网络情况,包含参数量、计算量
- 这是修改
后
调用的yolo.py
测试的yolov9-S-lwd.yaml
的打印网络情况,包含参数量、计算量
1.2 一些发现,YOLOv9代码打印的参数量计算量和Github上提供的并不一致,甚至yolov9-c.yaml代码打印出来是Github的两倍
1.3 开始创造YOLOv9-S
- 在
models/yolo.py
文件中,Ctrl+F
定位到args = [c1, c2, *args[1:]]
,然后再如下位置添加代码
# --------------------------添加的代码--------------------------- #