caffe python读取lmdb文件

版权声明: https://blog.csdn.net/b876144622/article/details/79972560
一般用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即可。
阅读更多

扫码向博主提问

keep_forward

非学,无以致疑;非问,无以广识
去开通我的Chat快问

没有更多推荐了,返回首页