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

Question

我最近刚刚发布了一个关于文件路径错误的问题，我认为我已经解决了这个问题......我拥有的文件夹包含 .jpg 文件。出于某种原因，我收到了一个新错误：

RuntimeError：在以下子文件夹中找到 0 个文件：C:\Users\Lyn\Desktop\UTKFaceDataSet\Train_Set 支持的扩展名有：.jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,.webp

这是我修改后的代码：

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

您需要以下格式的目录结构才能使用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)

我使用了两个类别（狗和猫），但也可以是一个。