如何将numpy数组转换为leveldb或lmdb形式

2024-05-19 09:14:33 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试将使用pylearn2创建的numpy数组转换为leveldb/lmdb,以便在Caffe中使用。
datasets/make_cifar10_gcn_whitened.py" rel="nofollow">This is the script我用来创建数据集的。 运行此脚本后,将生成两个文件,其中有test.pkltest.npytrain.pkltrain.npy
我不知道是否有一种直接的方法可以转换成leveldb/lmdb,所以假设没有办法,我需要能够读取每个图像及其对应的标签,这样我就可以将其保存到leveldb/lmdb数据库中。 有人告诉我需要使用pickle文件进行阅读,因为它提供了类似字典的结构。但是,试着

import cPickle as pickle
pickle.load( open( "N:/pylearn2-master/datasets/cifar10/pylearn2_gcn_whitened/test.pkl", "rb" ) )

输出
<pylearn2.datasets.cifar10.CIFAR10 at 0xde605f8> 我不知道访问pickle文件中的项的正确方法是什么,也不知道是否需要直接从numpy数组中读取。在


Tags: 文件testnumpylmdbtrain数组pickledatasets

热门问题