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

Question

我正在 Google colab 上使用 PyTorch 执行“猫与狗”图像分类任务（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

我遇到了这个问题，路径是

/content/usage

，然后我将数据文件夹移动到另一个名为数据集的文件夹，它可以工作！

mkdir dataset
!mv /content/usage/Casual dataset/Casual
!mv /content/usage/Ethnic dataset/Ethnic 
!mv /content/usage/Formal dataset/Formal
!mv /content/usage/Party dataset/Party
!mv /content/usage/Smart\ Casual dataset/Smart Casual
!mv /content/usage/Sports dataset/Sports
!mv /content/usage/Travel dataset/Travel

Answer 2

我遇到了同样的问题并刚刚解决，假设您想使用Google Colab对猫和鱼simple_example_link进行分类，您应该首先通过给定链接中的

download.py

下载图像（在单元格中运行此download.py） Colab 的），然后您将看到在 Colab 的左侧创建了 train、val 和 test 文件夹（参见附图），然后通常您只需将文件夹的路径添加到

ImageFolder 

正如我所做的那样。 如果您想安装驱动器，只需单击“安装驱动器”，如左下图所示