(self, inchannels, midchannels, outchannels, upfactor=2)
| 190 | |
| 191 | class FFM(nn.Module): |
| 192 | def __init__(self, inchannels, midchannels, outchannels, upfactor=2): |
| 193 | super(FFM, self).__init__() |
| 194 | self.inchannels = inchannels |
| 195 | self.midchannels = midchannels |
| 196 | self.outchannels = outchannels |
| 197 | self.upfactor = upfactor |
| 198 | |
| 199 | self.ftb1 = FTB(inchannels=self.inchannels, midchannels=self.midchannels) |
| 200 | # self.ata = ATA(inchannels = self.midchannels) |
| 201 | self.ftb2 = FTB(inchannels=self.midchannels, midchannels=self.outchannels) |
| 202 | |
| 203 | self.upsample = nn.Upsample(scale_factor=self.upfactor, mode='bilinear', align_corners=True) |
| 204 | |
| 205 | self.init_params() |
| 206 | |
| 207 | def forward(self, low_x, high_x): |
| 208 | x = self.ftb1(low_x) |
nothing calls this directly
no test coverage detected