YOLOv5添加注意力机制 Pytorch

Posted K同学啊

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了YOLOv5添加注意力机制 Pytorch相关的知识，希望对你有一定的参考价值。

我的YOLOv5版本为6.0

需要修改的两个文件如下：

  • ./models/yolo5s_SE-Net.yaml
  • ./models/commom.py

1. yolo5s_SE-Net.yaml文件

yolo5s_SE-Net.yaml 修改自文件yolo5s.yaml

yolo5s.yaml 文件内容为：

# YOLOv5 
以上是关于YOLOv5添加注意力机制 Pytorch的主要内容，如果未能解决你的问题，请参考以下文章
 

 Yolov5--从模块解析到网络结构修改（添加注意力机制）


 手把手带你Yolov5 (v6.1)添加注意力机制（二)（在C3模块中加入注意力机制）


 手把手带你Yolov5 (v6.1)添加注意力机制（并附上30多种顶会Attention原理图）


 YOLOv5-6.1添加注意力机制（SECBAMECACA）


 目标检测yolov5改进系列：主干网络中添加SE注意力机制网络


 yolov5改进之加入CBAM,SE,ECA,CA,SimAM,ShuffleAttention,Criss-CrossAttention,CrissCrossAttention多种注意力机制