PyTorch和Google Colab：ImageFolder不会加载所有图像

Question

我将PyTorch用于Google colab上的“猫与狗”的图像分类任务（9999张猫和9999张狗），并使用以下代码创建数据加载器：

import torchvision
from torchvision import transforms
data = torchvision.datasets.ImageFolder(r'/content/drive/My Drive/kaggle/trainset/trainset',transform = transforms.ToTensor())

和'数据'的长度为6594（预期为19998）。我尝试重新启动运行时，并且每次都获得不同数量的图像。然后，我尝试在jupyter笔记本中运行相同的代码，它按预期方式工作，加载了所有19998张图像。我在colab中做错什么了？

Answer 1

我遇到了同样的问题，刚刚解决了，假设您要使用Google Colab对猫和鱼simple_example_link进行分类，则应首先通过给定链接中的download.py下载图像（运行此在colab单元格中的download.py），然后您将看到火车，val和测试文件夹将在Colab的左侧创建（请参见附图），然后通常只需添加以下路径即可像我一样将文件夹保存到ImageFolder。 如果您要安装驱动器，只需单击“安装驱动器”，如左侧

下方的图片所示。