YOLOv8添加注意力机制

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中模块的代码，导致报错