pytorch ImageFolder数据不匹配原始imgs号吗？

Question

我有21,522件物品，共计609.7 MB在火车文件夹中

train
  -1(3600 png)
  -2(3600 png)
   .
   .
  -6

train_trans = transforms.Compose([
        vision.transforms.Resize(target_size),
        vision.transforms.RandomHorizontalFlip(),
        vision.transforms.RandomRotation(20),
        #CIFAR10Policy(),
        vision.transforms.ToTensor(),
        vision.transforms.Normalize(
            [0.485, 0.456, 0.406], 
            [0.229, 0.224, 0.225])
])
valid_trans = transforms.Compose([
        vision.transforms.Resize(target_size),
        vision.transforms.RandomResizedCrop(target_size, scale=(0.8,1.0)),
        vision.transforms.RandomHorizontalFlip(),
        vision.transforms.ToTensor(),
        vision.transforms.Normalize(
            [0.485, 0.456, 0.406], 
            [0.229, 0.224, 0.225])
])
test_trans = transforms.Compose([
        vision.transforms.Resize((size,size)),
        vision.transforms.RandomResizedCrop(target_size, scale=(0.8,1.0)),
        vision.transforms.ToTensor(),
        vision.transforms.Normalize(
            [0.485, 0.456, 0.406], 
            [0.229, 0.224, 0.225])
])

但是，当我加载数据集时

train_dataset = torchvision.datasets.ImageFolder(root= path + 'train/',transform=trans)

不知何故

print(len(train_dataset)) = 4109

不知怎么的，它没有加载整个图像文件。

Answer 1

如果您想要访问给定文件夹中的所有图像，则给定的路径应该在上面一个级别。

例如，当您将路径设为root= path + 'train/'时，ImageFolder只查看../train/文件夹中的文件夹，而不查看该文件夹中的图像文件。因此，要么在“列车”文件夹内的文件夹中组织文件，要么确保根文件夹中只有“列车”文件夹，并提供根文件夹的路径。