(self, in_channels)
| 792 | |
| 793 | class REM_decoder(nn.Module): |
| 794 | def __init__(self, in_channels): |
| 795 | super(REM_decoder, self).__init__() |
| 796 | self.decoder4 = Decoder4(in_channels) |
| 797 | self.decoder3 = Decoder(in_channels) |
| 798 | self.decoder2 = Decoder(in_channels) |
| 799 | self.decoder1 = Decoder(in_channels) |
| 800 | |
| 801 | def forward(self, x, prior_cam, pic): |
| 802 | f1, f2, f3, f4 = x |