I’m using torchvision ImgaeFolder class to create my dataset. I’m using a custom loader function.
By default ImageFolder creates labels according to different directories. I want to change this behaviour to custom one. Means I want to assign labels to each image. How can I do that ?