PyTorch - 加载没有子文件夹的图像

Question

首先我想说我是 PyTorch 的新手，所以如果我的问题水平不是那么高，我很抱歉。我想知道您是否可以帮我解决一些问题（我实际上有两个问题）。它们背后的故事：我正在研究图像分类。我的测试数据根据标签分为子文件夹，我通过 DataLoader 加载它们。第一个问题：1）如果你用特定的batch size训练你的模型，用其他sizes测试它会影响准确率吗？2) 有没有办法加载和使用带有位于单个文件夹（没有子文件夹）中的测试数据的模型。据我所知，DataLoader 需要子文件夹。

Answer 1

1. 这取决于您是否使用依赖于批次中其他项目的操作。如果您正在使用诸如批量标准化之类的东西，它可能会，但一般来说，如果您的网络单独处理批量项目，则不会。

2. 如果您查看torch.utils.data.Dataset的文档，您会发现数据集本质上只需要__len__和__getitem__方法，其中前者表示数据集包含多少个项目，后者获取第ith 个项目 - 无论是图像还是标签、图像及其分割掩码或其他东西。没有什么可以阻止您编写自定义数据集。我建议您查看DatasetFolder 的源代码并根据您的需要进行修改。