search

milesial / Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images
GNU General Public License v3.0
8.65k stars 2.4k forks source link

what does n_classes=1 mean? #493

Open coconut0713 opened 2 months ago

coconut0713 commented 2 months ago

For binary classification problems, I think n_classes should be 2 corresponding to mask_value={0, 1}. But when to use n_classes=1?