(self, in_channels)
| 844 | Output: the segmentation map from the current stage |
| 845 | ''' |
| 846 | def __init__(self, in_channels): |
| 847 | super(Decoder4_noEdge, self).__init__() |
| 848 | self.rcab_a = RCAB(in_channels) |
| 849 | self.rcab_b = RCAB(in_channels) |
| 850 | |
| 851 | self.conv3x3 = BasicConv2d(in_channels * 2, in_channels, kernel_size=3, padding=1) |
| 852 | self.out_s = nn.Conv2d(in_channels, 1, kernel_size=3, padding=1) |
| 853 | |
| 854 | def forward(self, f4, prior_cam): |
| 855 | 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