Hi,
I am currently using a concatenated dataset containing rgb images and depth images. So, I use something like
class ConcatDataset(torch.utils.data.Dataset):
def __init__(self, *datasets):
self.datasets = datasets
def __getitem__(self, i):
return tuple(d[i] for d in self.datasets)
def __len__(self):
return min(len(d) for d in self.datasets)
I am using the ImageFolder
for getting them. The rgb and depth image that I get are not corresponding ones(i.e) it is different and random from the folder.
What is the best way to get corresponding images? Kindly help me with this.
Thanks!!!