Kapil_Rana
(Kapil Rana)
February 25, 2020, 6:05am
1
train_dataset = torchvision.datasets.ImageFolder(
root=data_path,
transform=transform1
)
train_loader = torch.utils.data.DataLoader(
train_dataset,
batch_size=batch_size,
num_workers=0,
shuffle=False
)
train_dataset2 = torchvision.datasets.ImageFolder(
root=data_path,
transform=transform1
)
train_loader2 = torch.utils.data.DataLoader(
train_dataset2,
batch_size=batch_size,
num_workers=0,
shuffle=False
)
I want to use image data from these two dataloaders.
Kushaj
(Kushajveer Singh)
February 25, 2020, 6:11am
2
How do you want to use the data? You can call the two dataloaders independently in the same iteration.
train_iter = iter(train_loader)
train_iter2 = iter(train_loader2)
for epoch in range(10):
for iter in in range(10000):
data1 = train_iter.next()
data2 = train_iter2.next()
Kapil_Rana
(Kapil Rana)
February 25, 2020, 6:15am
3
I am feeding data to one single model for image classification which has 4 classes. I need data and targets from the dataloaders.
Kushaj
(Kushajveer Singh)
February 29, 2020, 10:19am
4
Did the above method work for you?
No,
Actually I am using ImageFolderLoader.