Deep learning Pyrotch ImageFolder数据大小与原始imgs编号不匹配？

我在train文件夹中有21522个项目，总计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)

不知怎的

打印（len（列车数据集））=4109

---

不知何故，它不会加载整个图像文件

如果要访问给定文件夹中的所有图像，则给定的路径应高于一级

---

例如，当您将路径指定为

root=path+'train/'

时，则

ImageFolder

仅查看

。/train/

文件夹中的文件夹，而不查看该文件夹中的图像文件。因此，可以将文件组织到train文件夹中的文件夹中，或者确保根文件夹中只有train文件夹，并提供根文件夹的路径

我不认为有任何方法可以用给出的信息来调试它。我唯一的猜测是，您文件夹中的图像可能没有所有可接受的扩展名，即其中之一：

IMG_扩展名=（“.jpg”、“.jpeg”、“.png”、“.ppm”、“.bmp”、“.pgm”、“.tif”、“.tiff”、“.webp”）