(self, num_classes=4)
| 62 | # Decoder (no pixel-wise MLP, no uncertainty-guided sampling) |
| 63 | class Decoder(nn.Module): |
| 64 | def __init__(self, num_classes=4): |
| 65 | super(Decoder, self).__init__() |
| 66 | self.conv2 = nn.Conv2d(2048, 2048, kernel_size=1, stride=1, padding=0) |
| 67 | self.up1 = UpSampleBN(skip_input=2048 + 176, output_features=1024) |
| 68 | self.up2 = UpSampleBN(skip_input=1024 + 64, output_features=512) |
| 69 | self.up3 = UpSampleBN(skip_input=512 + 40, output_features=256) |
| 70 | self.up4 = UpSampleBN(skip_input=256 + 24, output_features=128) |
| 71 | self.conv3 = nn.Conv2d(128, num_classes, kernel_size=3, stride=1, padding=1) |
| 72 | |
| 73 | def forward(self, features): |
| 74 | x_block0, x_block1, x_block2, x_block3, x_block4 = features[4], features[5], features[6], features[8], features[11] |
nothing calls this directly
no test coverage detected