Concat dataset

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)

train_loader = torch.utils.data.DataLoader(
             ConcatDataset( # concat
                 datasets.ImageFolder(traindir_A),
                 datasets.ImageFolder(traindir_B)
             ),
             batch_size=args.batch_size, shuffle=True,
             num_workers=args.workers, pin_memory=True)

for i, (input, target) in enumerate(train_loader):
    ...

Source: discuss.pytorch.org

Similar codes

5 votes193 views2 answers
Forgot django admin password
Python
0 vote187 views1 answer
Pandas cumulative mean
Python
3 votes221 views1 answer
Lambda with two columns pandas
Python
104 votes196 views6 answers
How to rename columns in Pandas DataFrame
Python
0 vote191 views9 answers
Word generator in python
Python
5 votes198 views2 answers
How to make a calculator
Python
2 votes185 views1 answer
Python pandas apply to one column
Python
8 votes222 views2 answers
Pandas dataframe get number of columns
Python
0 vote179 views1 answer
Pandas get attribute of object
Python
1 vote268 views2 answers
Pyqt5 message box
Python