Sorry, not transforms.Tensor it is transform.ToTensor to convert PIL Image or numpy.ndarray (H x W x C) in the range [0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0]
Can you explain a bit about how to deal with masks in case of multi-class segmentation. I am confused about this particularly. Currently my each of 32 channel has pixel value 0-1. and each channel of mask shows a particular class.
I’ll do that