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 vote220 views1 answer
Locate certificate path python
Python
0 vote255 views1 answer
Python array set key value
Python
11 votes288 views1 answer
Iis betwwen in python
Python
0 vote261 views1 answer
List of imported modules in python
Python
0 vote255 views2 answers
Send message from server to client python
Python
0 vote259 views1 answer
Macos youtube-dl unable to get local issuer certificate _ssl.c:1131
Python
0 vote312 views1 answer
SystemError: <class 'cv2.CascadeClassifier'> returned a result with an error set
Python
0 vote296 views3 answers
Converter python to c++ code
Python
-2 votes267 views1 answer
Django modelform datetime
Python
0 vote287 views1 answer
Parquet folder single df dataframe
Python