hi all
I am new in PyTorch and now I am working on semantic segmentation topics
I have a question and I hope I will find the answer
I have an image (RGB.PNG) and mask (contain indices of 5 classes [0, 1, 2, 3, 4]) but has three-channel (.PNG) so:
How can I deal with the mask as input (batch_size x H x W)?  can I take the first channel
ex:  mask = mask[:,:,0] or how ?