(self, x, prior_cam, pic)
| 1014 | self.decoder1 = Decoder_noRCAB(in_channels) |
| 1015 | |
| 1016 | def forward(self, x, prior_cam, pic): |
| 1017 | f1, f2, f3, f4 = x |
| 1018 | f4_s, f4_e, p4_s, p4_e = self.decoder4(f4, prior_cam) |
| 1019 | p4_s_out = F.interpolate(p4_s, size=pic.size()[2:], mode='bilinear') |
| 1020 | p4_e_out = F.interpolate(p4_e, size=pic.size()[2:], mode='bilinear') |
| 1021 | |
| 1022 | f3_s, f3_e, p3_s, p3_e = self.decoder3(f3, f4_s, f4_e, p4_s, p4_e) |
| 1023 | p3_s_out = F.interpolate(p3_s, size=pic.size()[2:], mode='bilinear') |
| 1024 | p3_e_out = F.interpolate(p3_e, size=pic.size()[2:], mode='bilinear') |
| 1025 | |
| 1026 | f2_s, f2_e, p2_s, p2_e = self.decoder2(f2, f3_s, f3_e, p3_s, p3_e) |
| 1027 | p2_s_out = F.interpolate(p2_s, size=pic.size()[2:], mode='bilinear') |
| 1028 | p2_e_out = F.interpolate(p2_e, size=pic.size()[2:], mode='bilinear') |
| 1029 | |
| 1030 | f1_s, f1_e, p1_s, p1_e = self.decoder1(f1, f2_s, f2_e, p2_s, p2_e) |
| 1031 | p1_s_out = F.interpolate(p1_s, size=pic.size()[2:], mode='bilinear') |
| 1032 | p1_e_out = F.interpolate(p1_e, size=pic.size()[2:], mode='bilinear') |
| 1033 | |
| 1034 | return prior_cam, p4_s_out, p3_s_out, p2_s_out, p1_s_out, p4_e_out, p3_e_out, p2_e_out, p1_e_out |
| 1035 | |
| 1036 | |
| 1037 | if __name__ == '__main__': |
nothing calls this directly
no outgoing calls
no test coverage detected