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

作者：花生_TL007 | 2024-04-24 15:46:43

踩

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

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

在这里插入图片描述
将找到的代码粘贴进来

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

在这里插入图片描述

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

在这里插入图片描述

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

在这里插入图片描述

在这里插入图片描述

运行成功！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/花生_TL007/article/detail/480207