(self, channel, reduction=4)
| 164 | |
| 165 | class eSEModule(nn.Module): |
| 166 | def __init__(self, channel, reduction=4): |
| 167 | super(eSEModule, self).__init__() |
| 168 | self.avg_pool = nn.AdaptiveAvgPool2d(1) |
| 169 | self.fc = nn.Conv2d(channel, channel, kernel_size=1, padding=0) |
| 170 | self.hsigmoid = Hsigmoid() |
| 171 | |
| 172 | def forward(self, x): |
| 173 | input = x |