一、本文介绍
作为入门性篇章,这里介绍了SimAM注意力在YOLOv8中的使用。包含SimAM原理分析,SimAM的代码、SimAM的使用方法、以及添加以后的yaml文件及运行记录。
二、SimAM原理分析
SimAM官方论文地址:SimAM文章
SimAM官方代码地址:SimAM源代码
SimAM注意力机制:是一种轻量级的自注意力机制,计算注意力权重时使用的是线性层而不是点积。其过程为:
输入张量 x (B, C, H, W) -> 均值计算 -> 差异平方 -> 归一化 -> 生成注意力图 -> Sigmoid -> 输出(B, C, H, W)
相关代码:
SimAM注意力的代码,如下。
class SimAM(