Pytorch-UNet icon indicating copy to clipboard operation
Pytorch-UNet copied to clipboard

i'm using mask image as gray scale. is there any idea to change input mask as grayscale?

Open DLbeginner opened this issue 2 years ago • 1 comments

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?

DLbeginner avatar Apr 25 '23 02:04 DLbeginner

You should be able to follow the n_classes == 1 code path where masks have one channel in [0, 1]

milesial avatar Jun 03 '23 21:06 milesial