python - 加载 .npy 文件作为 pytorch 的数据集

Question

我在 .npy 文件中预处理了数据，我们称其为原始数据的 X.npy 和标签的 Y.npy。它们被组织起来以匹配两个文件中的每个元素(来自 X 的第一个元素具有来自 Y 的第一个标签等)。如何使用 torch.utils.data.DataLoader 将其加载为数据集？我是 pytorch 的新手，任何帮助都会很有用。

Answer 1

您也可以使用 DatasetFolder，它基本上是 ImageFolder 的基础类。使用此类，您可以提供自己的文件扩展名和加载程序来加载示例。

def npy_loader(path):
    return torch.from_numpy(np.load(path))