(self, x, prior_cam, pic)
| 903 | self.decoder1 = Decoder_noSpade(in_channels) |
| 904 | |
| 905 | def forward(self, x, prior_cam, pic): |
| 906 | f1, f2, f3, f4 = x |
| 907 | f4_s, f4_e, p4_s, p4_e = self.decoder4(f4, prior_cam) |
| 908 | p4_s_out = F.interpolate(p4_s, size=pic.size()[2:], mode='bilinear') |
| 909 | p4_e_out = F.interpolate(p4_e, size=pic.size()[2:], mode='bilinear') |
| 910 | |
| 911 | f3_s, f3_e, p3_s, p3_e = self.decoder3(f3, f4_s, f4_e, p4_s, p4_e) |
| 912 | p3_s_out = F.interpolate(p3_s, size=pic.size()[2:], mode='bilinear') |
| 913 | p3_e_out = F.interpolate(p3_e, size=pic.size()[2:], mode='bilinear') |
| 914 | |
| 915 | f2_s, f2_e, p2_s, p2_e = self.decoder2(f2, f3_s, f3_e, p3_s, p3_e) |
| 916 | p2_s_out = F.interpolate(p2_s, size=pic.size()[2:], mode='bilinear') |
| 917 | p2_e_out = F.interpolate(p2_e, size=pic.size()[2:], mode='bilinear') |
| 918 | |
| 919 | f1_s, f1_e, p1_s, p1_e = self.decoder1(f1, f2_s, f2_e, p2_s, p2_e) |
| 920 | p1_s_out = F.interpolate(p1_s, size=pic.size()[2:], mode='bilinear') |
| 921 | p1_e_out = F.interpolate(p1_e, size=pic.size()[2:], mode='bilinear') |
| 922 | |
| 923 | 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 |
| 924 | |
| 925 | class REM_decoder_noSpade_noEdge(nn.Module): |
| 926 | def __init__(self, in_channels): |
nothing calls this directly
no outgoing calls
no test coverage detected