def __init__(self, n_channels, n_classes):
    super().__init__()
    self.inc = inconv(n_channels, 64)
    self.down1 = down(64, 128)
    self.down2 = down(128, 256)
    self.down3 = down(256, 512)
    self.down4 = down(512, 512)
    self.up1 = up(1024, 256)
    self.up2 = up(512, 128)
    self.up3 = up(256, 64)
    self.up4 = up(128, 64)
    self.outc = outconv(64, n_classes)
def __init__(self, n_channels, n_classes):
    super().__init__()
    self.inc = inconv(n_channels, 64)
    self.down1 = down(64, 128)
    self.down2 = down(128, 256)
    self.down3 = down(256, 512)
    self.down4 = down(512, 512)
    self.up1 = up(1024, 256)
    self.up2 = up(512, 128)
    self.up3 = up(256, 64)
    self.up4 = up(128, 64)
    self.outc = outconv(64, n_classes)