| 163 | |
| 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 |
| 174 | x = self.avg_pool(x) |
| 175 | x = self.fc(x) |
| 176 | x = self.hsigmoid(x) |
| 177 | return input * x |
| 178 | |
| 179 | |
| 180 | class _OSA_module(nn.Module): |