YOLOv8添加注意力机制_yolov8添加stn注意力机制

1、找好要添加的注意力机制的代码

2、放入注意力机制代码

主要是在ultralytics/nn/modules下的conv.py文件中添加注意力机制相关代码  

3、添加注意力机制模块

3.1 在conv.py中添加

然后还需要在conv.py文件开头的all元组中添加注意力机制类名，方便引用

 3.2 在init.py中添加

 然后还需要到init.py文件中进行注意力机制类名的添加

3.3 在task.py中添加

4、添加YOLOv8处理对应模块代码

5、修改配置文件

6、可能出现如下所示问题

这里主要是因为引用的是pip install ultralytics中模块的代码，导致报错