PyTorch&；Google Colab:ImageFolder不'；不要加载所有图像

我使用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中做错了什么？

我遇到了同样的问题，刚刚解决了，假设你想用谷歌colab对猫和鱼进行分类，你应该首先通过给定链接中的

download.py

下载图像（在colab的一个单元格中运行这个download.py），然后你就会看到火车，val和测试文件夹将在Colab的左侧创建（请参见附件中的图片），然后通常您只需像我所做的那样将文件夹的路径添加到

ImageFolder

。

---

如果你想挂载你的驱动器，只需点击挂载驱动器，如左面下方的图片所示

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

download.py

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

ImageFolder

。

---

如果你想挂载你的驱动器，只需点击挂载驱动器，如左面下方的图片所示

我会先从

/content/drive/

中复制数据。这样行吗？我会先从

/content/drive/

中复制数据。行吗？