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

1 vote116 views1 answer
Can variables have spaces python
Python
0 vote112 views1 answer
Python get_loc not returning number
Python
186 votes188 views6 answers
Pip upgrade command
Python
0 vote136 views1 answer
Make blinking text python1
Python
0 vote111 views1 answer
Queue using linked list in python
Python
-1 vote111 views1 answer
Img_sm = pygame.transform.scale(img, (32, 32))
Python
0 vote101 views1 answer
File "script.py", line 1 import module math ^ SyntaxError: invalid syntax
Python
0 vote112 views1 answer
Opencv waitkey example
Python
0 vote102 views1 answer
Operator.itemgetter(1) in python
Python
0 vote142 views1 answer
1. write a program to multiply two numbers using function python
Python