(self, num_classes, channels)
| 232 | |
| 233 | class Exp2_Decoder12(nn.Module): |
| 234 | def __init__(self, num_classes, channels): |
| 235 | super().__init__() |
| 236 | channels8,channels16=channels["8"],channels["16" ] |
| 237 | self.head16=ConvBnAct(channels16, 128, 1) |
| 238 | self.head8=ConvBnAct(channels8, 128, 1) |
| 239 | self.conv=ConvBnAct(128,128,1) |
| 240 | self.classifier=nn.Conv2d(128, num_classes, 1) |
| 241 | |
| 242 | def forward(self, x): |
| 243 | #intput_shape=x.shape[-2:] |