YOLOv8创新魔改教程(二)如何添加注意力机制
(一)找代码
github找各种注意力机制的代码
(二)融合
1.创建文件
在ultralytics/nn/attention.py创建attention.py 文件

将找到的代码粘贴进来

2.修改task.py文件
打开 ultralytics/nn/tasks.py ,将注意力机制导入,名称就是注意力机制类的名称


在tasks.py文件中def parse_model函数,加上这段代码(需要接受通道数的加,不需要的不用加)

3.修改tyaml文件
创建一个yolov8-attention.yaml文件,将注意力机制添加上,注意修改网络层数

(三)运行

运行成功!