| 33 | |
| 34 | |
| 35 | class to_map(nn.Module): |
| 36 | def __init__(self,channels): |
| 37 | super(to_map,self).__init__() |
| 38 | self.to_map = nn.Sequential( |
| 39 | nn.Conv2d(in_channels=channels,out_channels=1, kernel_size=1,stride=1), |
| 40 | nn.Sigmoid() |
| 41 | ) |
| 42 | |
| 43 | def forward(self,feature): |
| 44 | return self.to_map(feature) |
| 45 | |
| 46 | |
| 47 | class conv_bn_relu(nn.Module): |