| 264 | |
| 265 | class GCM_interFA(nn.Module): |
| 266 | def __init__(self, in_channels, out_channels): |
| 267 | super(GCM_interFA, self).__init__() |
| 268 | self.T1 = ETM(in_channels, out_channels) |
| 269 | self.T2 = ETM(in_channels * 2, out_channels) |
| 270 | self.T3 = ETM(in_channels * 4, out_channels) |
| 271 | self.T4 = ETM(in_channels * 8, out_channels) |
| 272 | self.interFA1 = InterFA(out_channels) |
| 273 | self.interFA2 = InterFA(out_channels) |
| 274 | self.interFA3 = InterFA(out_channels) |
| 275 | |
| 276 | def forward(self, f1, f2, f3, f4): |
| 277 | f1 = self.T1(f1) # 96,96,96 |