MCPcopy Create free account
hub / github.com/ChunmingHe/WS-SAM / __init__

Method __init__

lib/Modules.py:953–959  ·  view source on GitHub ↗
(self, in_channels)

Source from the content-addressed store, hash-verified

951
952class Decoder4_noRCAB(nn.Module):
953 def __init__(self, in_channels):
954 super(Decoder4_noRCAB, self).__init__()
955 self.crb = BasicConv2d(in_channels, in_channels, kernel_size=3, padding=1)
956 self.out_e = nn.Conv2d(in_channels, 1, kernel_size=3, padding=1)
957
958 self.conv3x3 = BasicConv2d(in_channels * 2, in_channels, kernel_size=3, padding=1)
959 self.out_s = nn.Conv2d(in_channels, 1, kernel_size=3, padding=1)
960
961 def forward(self, f4, prior_cam):
962 prior_cam = F.interpolate(prior_cam, size=f4.size()[2:], mode='bilinear', align_corners=True) # 2,1,12,12->2,1,48,48

Callers

nothing calls this directly

Calls 2

BasicConv2dClass · 0.85
__init__Method · 0.45

Tested by

no test coverage detected