pytorch torchvision.datasets.ImageFolder FileNotFoundError：找不到类 .ipynb_checkpoints 的有效文件

Question

尝试在 Colab 中使用 pytorch torch.datasets.ImageFolder 加载训练数据。

transform = transforms.Compose([transforms.Resize(400),
                                transforms.ToTensor()])
dataset_path = 'ss/'
dataset = datasets.ImageFolder(root=dataset_path, transform=transform)
dataloader = torch.utils.data.DataLoader(dataset, batch_size=20)

我遇到了以下错误：

---------------------------------------------------------------------------
FileNotFoundError                         Traceback (most recent call last)
<ipython-input-27-7abcc1f434b1> in <module>()
      2                                 transforms.ToTensor()])
      3 dataset_path = 'ss/'
----> 4 dataset = datasets.ImageFolder(root=dataset_path, transform=transform)
      5 dataloader = torch.utils.data.DataLoader(dataset, batch_size=20)

3 frames
/usr/local/lib/python3.7/dist-packages/torchvision/datasets/folder.py in make_dataset(directory, class_to_idx, extensions, is_valid_file)
    100         if extensions is not None:
    101             msg += f"Supported extensions are: {', '.join(extensions)}"
--> 102         raise FileNotFoundError(msg)
    103 
    104     return instances

FileNotFoundError: Found no valid file for the classes .ipynb_checkpoints. Supported extensions are: .jpg, .jpeg, .png, .ppm, .bmp, .pgm, .tif, .tiff, .webp

我的数据集文件夹包含一个子文件夹，其中包含许多 png 格式的训练图像，但 ImageFolder 仍然无法访问它们。

Answer 1

我在使用类似IPython笔记本的工具时遇到了同样的问题。

首先请检查一下你的.txt文件下是否有隐藏文件dataset_path。ls -a如果您在Linux环境下使用。

我遇到的情况是我发现了一个名为的隐藏文件，.ipynb_checkpoints该文件与图像类子文件夹并行。我认为该文件会导致 PyTorch 数据集混乱。我确定它没有用，所以我就把它删除了。然后数据集就可以正常工作了。

或者，如果您想简单地忽略该文件，您也可以尝试这个。