logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

SEnet、CBAM、ECA注意力三大机制

# 下面是通道注意力机制即SEnet的代码,个人理解SEnet是在对每个channel进行加权操作,看那个channel重要,就给他权重大一点。# 每个channel由于每个特征向量不一样因此提取的特征也就不一样即通道注意力机制注重的是特征的重要性(图片的点、线、明暗)# 因此通道注意力机制是对每个通道的长宽进行卷积,一开始对通道不进行改变import torchimport torch.nn a

#pytorch#深度学习#计算机视觉 +2
暂无文章信息