(self, inchannels, reduction=8)
| 146 | |
| 147 | class ATA(nn.Module): |
| 148 | def __init__(self, inchannels, reduction=8): |
| 149 | super(ATA, self).__init__() |
| 150 | self.inchannels = inchannels |
| 151 | self.avg_pool = nn.AdaptiveAvgPool2d(1) |
| 152 | self.fc = nn.Sequential(nn.Linear(self.inchannels * 2, self.inchannels // reduction), |
| 153 | nn.ReLU(inplace=True), |
| 154 | nn.Linear(self.inchannels // reduction, self.inchannels), |
| 155 | nn.Sigmoid()) |
| 156 | self.init_params() |
| 157 | |
| 158 | def forward(self, low_x, high_x): |
| 159 | n, c, _, _ = low_x.size() |
nothing calls this directly
no test coverage detected