I am working on image segmentation, on Cityscape dataset, and whenever i uses more than 7 classes torchvision doesn’t save correct coloured image. however when I tried to save it using numpy and PIL it was saved as expected. can someone explain what am I missing here.
Thanks in advance.