Squeeze-and-Excitation Networks(SENet)
结构和代码如下(参考:b站视频:YOLOv5 v6.1添加SE,CA,CBAM,ECA注意力机制教学,即插即用):
Global pooling:每个channel上面的所有点做平均,这样每个channel都输出一个数。所以左图中,HxWxC变成了1x1xC。(参考:关于global average pooling理解和介绍)
FC就是全连接层。
结构和代码如下(参考:b站视频:YOLOv5 v6.1添加SE,CA,CBAM,ECA注意力机制教学,即插即用):
Global pooling:每个channel上面的所有点做平均,这样每个channel都输出一个数。所以左图中,HxWxC变成了1x1xC。(参考:关于global average pooling理解和介绍)
FC就是全连接层。