I am trying to create a dataset with an imagefolder as input and another imagefolder to use as labels. Is there any way I can have them both in the same DataLoader?
My code is as follows:
dataset = torchvision.datasets.ImageFolder(root=TRAIN_DATA_PATH, transform=TRANSFORM_IMG)
dataset_labels = torchvision.datasets.ImageFolder(root=TRAIN_LABELS_PATH, transform=TRANSFORM_IMG)

I would like to be able to integrate dataset_labels into dataset, so that I can then just create a DataLoader from dataset.