从文件夹导入 JPG 文件以进行 PyTorch 转换（另一个错误）？

Question

I just recently posted a question about a file path error in which I thought I resolved... the folder that I have contains.jpg files. For some reason, i'm getting a new error:

RuntimeError: Found 0 files in subfolders of: C:\Users\Lyn\Desktop\UTKFaceDataSet\Train_Set Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,.webp

Here is my modified code:

data_dir = "C:\\Users\\Lyn\\Desktop\\UTKFaceDataSet"
train_dir = data_dir + '\\Train_Set'
test_dir = data_dir + '\\Test_Set'

training_transforms = transforms.Compose([transforms.Resize(100),
                                          transforms.ToTensor()])
testing_transforms = transforms.Compose([transforms.Resize(256),
                                         transforms.ToTensor()])

#Load the datasets with ImageFolder
training_dataset = datasets.ImageFolder(train_dir, transform=training_transforms)
testing_dataset = datasets.ImageFolder(test_dir, transform=testing_transforms)

Answer 1

You need your directory structure in the following format to use the datasets.ImageFolder() :

dataset
   train_dataset
      dogs
         a.png, b.png
      cats
         c.png, d.png
   valid_dataset
      dogs
         e.png, f.png
      cats
         g.png, h.png

train_dir = 'dataset/train_dataset/'
training_dataset = datasets.ImageFolder(train_dir, transform=training_transforms)

I used two categories (dogs and cats), but it could be one also.