查看lmdb中的数据

lmdb是caffe钦定的数据库类型,有时候难免需要查看lmdb数据库中的数据,方法如下:

1
2
3
4
5
6
7
8
9
10
11
import lmdb
import caffe
lmdbpath='/data2/guoyu/rmppe/Realtime_Multi-Person_Pose_Estimation/training/COCO_kpt/lmdb'
datum = caffe.proto.caffe_pb2.Datum()
env=lmdb.open(lmdbpath)
txn=env.begin()
cursor=txn.cursor()
for key,value in cursor:
datum.ParseFromString(value)
data=caffe.io.datum_to_array(datum)
print data[3,:,:]