caffe python读取lmdb文件

一般用caffe训练模型的时候，都会首先把数据写入数据库，lmdb/leveldb， 但如果写入之后想check一下是否写对了，怎么去读lmdb文件呢？
这里用python代码读取lmdb文件，查看其中的内容，看是否符合预期。


代码如下：


-Python 代码
01
import caffe
02
import lmdb
03
 
04
lmdb_env = lmdb.open('train_lmdb') #caffe保存lmdb文件的目录
05
lmdb_txn = lmdb_env.begin()
06
lmdb_cursor = lmdb_txn.cursor()
07
datum = caffe.proto.caffe_pb2.Datum()
08
 
09
for key, value in lmdb_cursor:
10
    datum.ParseFromString(value)
11
    label = datum.label
12
    data = caffe.io.datum_to_array(datum)
13
    print label, data


这里导入了lmdb和caffe两个库，解释一下如何导入


1、lmdb，需要用pip安装lmdb，进入root账户，输入 pip install lmdb即可(在其它用户下，输入sudo命令无法安装成功)


2、import caffe， 安装好caffe后，在bashrc中，添加环境变量， export PYTHONPYTH="xxx/caffe-master/python", 再source ~/.bashrc


import caffe的时候，会报一个错，是因为没有编译pycaffe，在caffe-master下，make pycaffe即可。