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

我已经对.npy文件中的数据进行了预处理，对于原始数据，我们将其称为X.npy；对于标签，将其称为Y.npy。它们经过组织以匹配两个文件中的每个元素（X中的第一个元素具有Y中的第一个标签，依此类推）。如何使用torch.utils.data.DataLoader将其加载为数据集？我是pytorch的新手，任何帮助都将非常有用。

最佳答案

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

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

关于python - 加载.npy文件作为pytorch的数据集，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/57989716/