#YOLO# #目标检测# #计算机视觉#
一、本文介绍
作为入门性篇章,这里介绍了GAM注意力在YOLOv8中的使用。包含GAM原理分析,GAM的代码、GAM的使用方法、以及添加以后的yaml文件及运行记录。
二、GAM原理分析
GAM官方论文地址:文章
GAM官方代码地址:
GAM注意力机制:GAM采用了顺序的通道-空间注意力机制(类似CBAM),但对子模块进行了重新设计。
三、相关代码:
GAM注意力的代码,如下:
class GAM(nn.Module):
def __init__(self, in_channels, rate=4):
super().__init__()
out_channels