search

bigmb / Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
MIT License
1.87k stars 345 forks source link

hello,is label a 3 channels image? #51

Closed CasonTsai closed 3 years ago

CasonTsai commented 3 years ago

i got this mistake ,it the label a 3 channels image?not a single channle image image

CasonTsai commented 3 years ago

my input image is 3 channels,label image is single channel

bigmb commented 3 years ago

Hello @CasonTsai ,

The input image should be 3 channels and the label should be 1 channel. Please have a look at this issue [(https://github.com/bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets/issues/21)]

Also have a look at [https://github.com/bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets/blob/master/Data_Loader.py]

CasonTsai commented 3 years ago

ok,thanks

Alex28132 commented 10 months ago

my input image is 3 channels,label image is single channel

Sir, have you solved this problem? I have also encountered this problem