UNET Multiclass Segmentation from Binary Segmentation ...
discuss.pytorch.org › t › unet-multiclassJun 07, 2021 · Hi, Any tips on how I would modify the following UNET binary segmentation architecture for multi-class segmentation with 6 classes? I am new to PyTorch and I am lost with regards to modifying this network for a multi-class usage. Mainly, I do not understand how the two architectures would differ. Thank you! class DoubleConv(nn.Module): def __init__(self, in_channels, out_channels): super ...