when i use imagefolder,it had error:
Traceback (most recent call last):
File “drive/gan/main.py”, line 172, in
File “drive/gan/main.py”, line 59, in train
dataset = tv.datasets.ImageFolder(opt.data_path,transform=transforms)
File “/usr/local/lib/python3.6/dist-packages/torchvision/datasets/folder.py”, line 103, in init
"Supported image extensions are: " + “,”.join(IMG_EXTENSIONS)))
RuntimeError: Found 0 images in subfolders of: drive/ganData/
Supported image extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm
there is the code:
transforms = tv.transforms.Compose([
tv.transforms.ToTensor(),
tv.transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
])
dataset = tv.datasets.ImageFolder('drive/ganData/',transform=transforms)
dataloader = t.utils.data.DataLoader(dataset,
batch_size = opt.batch_size,
shuffle = True,
num_workers= opt.num_workers,
drop_last=True
)
there is the data: