传通道的注意力添加
不传通道的注意力添加
1在ultralytics/nn下复制注意力机制代码
2在task.py中导入注意力代码
3在def parse_model中增加通道传入
elif m is CBAM:
args = [ch[f], *args]
4在模型配置文件yolov8x.yaml中添加注意力
- [-1, 1, ShuffleAttention, [16,8]]
1在ultralytics/nn下复制注意力机制代码
2在task.py中导入注意力代码
3在def parse_model中增加通道传入
elif m is CBAM:
args = [ch[f], *args]
4在模型配置文件yolov8x.yaml中添加注意力
- [-1, 1, ShuffleAttention, [16,8]]