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

luaexgnf 于 9个月前发布在其他

关注(0)|答案(2)|浏览(83)

我在.npy文件中有预处理的数据，让我们称之为原始数据的X.npy和标签的Y.npy。它们被组织以匹配两个文件中的每个元素（X中的第一个元素具有Y中的第一个标签等）。我如何使用torch.utils.data.DataLoader将其加载为数据集？我对pytorch非常陌生，任何帮助都将是有用的。

numpy

来源：https://stackoverflow.com/questions/57989716/loading-npy-files-as-dataset-for-pytorch

2条答案

按热度按时间

nxagd54h1#

你也可以使用DatasetFolder，它基本上是ImageFolder的底层类。使用这个类你可以提供你自己的文件扩展名和加载器来加载样本。

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

字符串

赞(0）回复(0）举报 9个月前

tzxcd3kk2#

所有代码如下所列：

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

dataset = datasets.DatasetFolder(
    root='PATH',
    loader=npy_loader,
    extensions=['.npy']
)

字符串

赞(0）回复(0）举报 9个月前

我来回答

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

2条答案

相关问题

热门标签

最新问答