(self, num_classes,channels)
| 212 | |
| 213 | class Exp2_Decoder10(nn.Module): |
| 214 | def __init__(self, num_classes,channels): |
| 215 | super().__init__() |
| 216 | channels8,channels16=channels["8"],channels["16"] |
| 217 | self.head8=ConvBnAct(channels8,32,1) |
| 218 | self.head16=ConvBnAct(channels16,128,1) |
| 219 | self.conv=DBlock(128+32,128,[1],16,1,"se") |
| 220 | self.classifier=nn.Conv2d(128, num_classes, 1) |
| 221 | |
| 222 | def forward(self, x): |
| 223 | #intput_shape=x.shape[-2:] |