YOLO添加注意力机制(SE、CBAM、ECA、CA)
上一章 【YOLO】自定义训练数据集
文章目录
前言
注意力机制(attention mechanism)受人类视觉机制的研究,让网络学习更关注的特征。注意力机制主要有:自注意力(self-attention)、软注意力(soft-attention)、硬注意力(hard-attention)。其已经广泛应用于分类、检测、分割等任务。
一、SE(SENet)
1.1 原理
论文地址:https://arxiv.org/pdf/1709.01507.pdf
github地址:https://github.com/hujie-frank/SENet
卷积神经网络 (CNN) 的核心构建块是卷积算子