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

0 vote139 views1 answer
How to check if a function false python
Python
0 vote144 views1 answer
Import one file into another python
Python
1 vote134 views1 answer
Findfont: Font family ['Times New Roman'] not found. Falling back to DejaVu Sans. findfont: Font family ['Times New Roman'] not found. Falling back to DejaVu Sans.
Python
3 votes138 views2 answers
Python decouple
Python
0 vote146 views1 answer
To text pandas
Python
0 vote135 views1 answer
Python version of settimout
Python
0 vote144 views1 answer
How to give a role permissions discord py
Python
17 votes160 views3 answers
Set title matplotlib
Python
5 votes144 views4 answers
While true loop python
Python
0 vote135 views2 answers
Convert string to list of dictionaries
Python