论文传送门:Squeeze-and-Excitation Networks
SENet的目的:
让网络对特征层的某些重要的通道更加关注。
SENet的方法:
使用SEblock通道注意力机制模块。
SEblock的结构:
SEblock的使用:
SEblock可以作为一个即插即用的模块,理论上可以添加到任意特征层后。原文中给出其在Inception Module和Residual Module中的使用方法。
import torch
import torch.nn as nn
论文传送门:Squeeze-and-Excitation Networks
让网络对特征层的某些重要的通道更加关注。
使用SEblock通道注意力机制模块。
SEblock可以作为一个即插即用的模块,理论上可以添加到任意特征层后。原文中给出其在Inception Module和Residual Module中的使用方法。
import torch
import torch.nn as nn