我在 .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/