python lmdb demo 这接口和BDB一样恶心啊！

Posted 2020-09-19

import lmdb

lmdb_img_name = "test.lmdb"
env = lmdb.open(lmdb_img_name, map_size=1e6)

with env.begin(write=True) as txn:
    txn.put("key", "xxxxxxxxxxxxxxxxxx")
    txn.put("key2", "?")
    txn.put("key3", "some value")
    #txn.commit()
env.close()


env = lmdb.open(lmdb_img_name, readonly=True)
with env.begin() as txn:
    print txn.get(b‘key‘)

    cursor = txn.cursor()
    for key, value in cursor:
        print(key, value)

 

更多见：http://blog.csdn.net/ayst123/article/details/44077903，摘录如下：

 

write

db_img = lmdb.Environment(lmdb_img_name,map_size=int(1e12))
txn_img = db_img.begin(write=True,buffers=True)
datum_img = getImgDatum(img_path)
txn_img.put(key,datum_img.SerializeToString())
txn_img.commit()
db_img.close()

这里，先建立lmdb的instance， db_img。 然后开始txn(可能有很多个开始？)。 put 存放，最后一定注意要commit()。不然根本没有存进去，检查下生成的lmdb文件就能看出。最后，close()

这里生成datum是用caffe自带的函数，做io.py里，叫caffe.io.array_to_datum, 将array变成datum.

Note:

datum存的时候一定要用datum.SerializeToString(), 将它变成string()

分段write

for ind,obj in enumerate(list):
    if ind%100=0:
        txn.commit()
        txn = env.begin(write=True)

txn.commit()
env.close()

每一次commit 后，需要再定义一次 txn=env.begin(write=True) 
for loop 外面最后再 commit

Read

img_lmdb = imdb.open("name")
txn= img_lmdb.begin()
cursor = txn.cursor()
cursor.iternext()
key = cursor.key()
value = cursor.value()

datum = caffe_pb2.Datum()
datum.ParseFromString(value)

Note:

cursor 第一个是空的， iternext()后是第一个值

循环

txn = … 
cursor = txn.cursor() 
for ( idx, (key, value) ) in enumerate(cursor): 
…………….

这就可以循环找了