(self, in_channels)
| 756 | |
| 757 | class Decoder(nn.Module): |
| 758 | def __init__(self, in_channels): |
| 759 | super(Decoder, self).__init__() |
| 760 | self.rcab_a = RCAB(in_channels) |
| 761 | self.rcab_b = RCAB(in_channels) |
| 762 | self.spade_a = Spade(in_channels, in_channels) |
| 763 | self.spade_b = Spade(in_channels, in_channels) |
| 764 | |
| 765 | self.crb = BasicConv2d(in_channels*2, in_channels, kernel_size=3, padding=1) |
| 766 | self.out_e = nn.Conv2d(in_channels, 1, kernel_size=3, padding=1) |
| 767 | |
| 768 | self.conv3x3 = BasicConv2d(in_channels * 2, in_channels, kernel_size=3, padding=1) |
| 769 | self.out_s = nn.Conv2d(in_channels, 1, kernel_size=3, padding=1) |
| 770 | |
| 771 | def forward(self, f, f_s, f_e, p_s, p_e): |
| 772 | prior_cam = F.interpolate(p_s, size=f.size()[2:], mode='bilinear', align_corners=True) # 2,1,12,12->2,1,48,48 |
nothing calls this directly
no test coverage detected