YOLOv8创新魔改教程(二)如何添加注意力机制

YOLOv8创新魔改教程(二)如何添加注意力机制

(一)找代码

github找各种注意力机制的代码

(二)融合

1.创建文件

在ultralytics/nn/attention.py创建attention.py 文件

YOLOv8创新魔改教程(二)如何添加注意力机制_第1张图片
将找到的代码粘贴进来
YOLOv8创新魔改教程(二)如何添加注意力机制_第2张图片

2.修改task.py文件

打开 ultralytics/nn/tasks.py ,将注意力机制导入,名称就是注意力机制类的名称

YOLOv8创新魔改教程(二)如何添加注意力机制_第3张图片
YOLOv8创新魔改教程(二)如何添加注意力机制_第4张图片

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

YOLOv8创新魔改教程(二)如何添加注意力机制_第5张图片

3.修改tyaml文件

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

YOLOv8创新魔改教程(二)如何添加注意力机制_第6张图片

(三)运行

YOLOv8创新魔改教程(二)如何添加注意力机制_第7张图片

运行成功!

你可能感兴趣的:(YOLOv8创新魔改,YOLO,注意力机制)