加载.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))