| 4 | from resnet import resnet50, resnet18 |
| 5 | |
| 6 | class Normalize(nn.Module): |
| 7 | def __init__(self, power=2): |
| 8 | super(Normalize, self).__init__() |
| 9 | self.power = power |
| 10 | |
| 11 | def forward(self, x): |
| 12 | norm = x.pow(self.power).sum(1, keepdim=True).pow(1. / self.power) |
| 13 | out = x.div(norm) |
| 14 | return out |
| 15 | |
| 16 | class Non_local(nn.Module): |
| 17 | def __init__(self, in_channels, reduc_ratio=2): |