Pytorch-UNet
Pytorch-UNet copied to clipboard
i'm using mask image as gray scale. is there any idea to change input mask as grayscale?
the original code is for rgb mask images.
but i'm using grayscale mask images as grayscale.
do i need to convert mask image or read images as grayscale?
You should be able to follow the n_classes == 1 code path where masks have one channel in [0, 1]