首页 > 代码库 > python lmdb demo 这接口和BDB一样恶心啊!

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

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(bkey)

    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 = http://www.mamicode.com/cursor.value()>

Note:

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

循环

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

这就可以循环找了

 

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