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文件,将注意力机制添加上,注意修改网络层数
(三)运行
运行成功!