(self, in_channels)
| 709 | |
| 710 | class Decoder4(nn.Module): |
| 711 | def __init__(self, in_channels): |
| 712 | super(Decoder4, self).__init__() |
| 713 | self.rcab_a = RCAB(in_channels) |
| 714 | self.rcab_b = RCAB(in_channels) |
| 715 | self.crb = BasicConv2d(in_channels, in_channels, kernel_size=3, padding=1) |
| 716 | self.out_e = nn.Conv2d(in_channels, 1, kernel_size=3, padding=1) |
| 717 | |
| 718 | self.conv3x3 = BasicConv2d(in_channels * 2, in_channels, kernel_size=3, padding=1) |
| 719 | self.out_s = nn.Conv2d(in_channels, 1, kernel_size=3, padding=1) |
| 720 | |
| 721 | def forward(self, f4, prior_cam): |
| 722 | prior_cam = F.interpolate(prior_cam, size=f4.size()[2:], mode='bilinear', align_corners=True) # 2,1,12,12->2,1,48,48 |
nothing calls this directly
no test coverage detected