Hi,
I am trying to train PSPNet (Pyramid Scene Parsing Network) for semantic segmentation. I have created the model but I am having difficulties in loading the dataset as I only have experience loading images with folder names as labels for image classification.
Will appreciate some help here to guide me on how to load training images and ground truth labels in PyTorch.
Thank you