一、本文介绍
作为入门性第一篇,这里介绍了ECA注意力在YOLOv8中的使用。包含ECA原理分析,ECA的代码、ECA的使用方法、以及添加以后的yaml文件及运行记录。
二、ECA原理分析
ECA官方论文地址:ECA文章
ECA的pytorch版代码:ECA的pytorch代码
ECA注意力机制:深度卷积神经网络的高效通道注意力机制。ECA机制是对SE网络进行了改进,将SE中使用的FC层,改为1×1卷积学习通道注意信息。感觉一新的得算是这个自适应卷积核的大小的处理方法:其ECA模块的原理结构如下图所示。
相关代码:
ECA注意力的代码,如下。
class ECA(