大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。
Jetbrains全系列IDE稳定放心使用
文章作者:Tyan
博客:noahsnail.com | CSDN | 简书
由于Caffe使用的存储图像的数据库是lmdb,因此有时候需要对lmdb文件进行操作,本文主要讲解如何用Python合并lmdb文件。没有lmdb支持的,需要用pip命令安装。
pip install lmdb
代码及注释如下:
# coding=utf-8
# filename: merge_lmdb.py
import lmdb
# 将两个lmdb文件合并成一个新的lmdb
def merge_lmdb(lmdb1, lmdb2, result_lmdb):
print 'Merge start!'
# env代表Environment, txn代表Transaction
# 打开lmdb文件,读模式
env_1 = lmdb.open(lmdb1)
env_2 = lmdb.open(lmdb2)
# 创建事务
txn_1 = env_1.begin()
txn_2 = env_2.begin()
# 打开数据库
database_1 = txn_1.cursor()
database_2 = txn_2.cursor()
# 打开lmdb文件,写模式,
env_3 = lmdb.open(result_lmdb, map_size=int(1e12))
txn_3 = env_3.begin(write=True)
count = 0
# 遍历数据库
for (key, value) in database_1:
# 将数据放到结果数据库事务中
txn_3.put(key, value)
count++
if(count % 1000 == 0):
# 将数据写入数据库,必须的,否则数据不会写入到数据库中
txn_3.commit()
count = 0
txn_3 = env_3.begin(write=True)
if(count % 1000 != 0):
txn_3.commit()
count = 0
txn_3 = env_3.begin(write=True)
for (key, value) in database_2:
txn_3.put(key, value)
if(count % 1000 == 0):
txn_3.commit()
count = 0
txn_3 = env_3.begin(write=True)
if(count % 1000 != 0):
txn_3.commit()
count = 0
txn_3 = env_3.begin(write=True)
# 关闭lmdb
env_1.close()
env_2.close()
env_3.close()
print 'Merge success!'
# 输出结果lmdb的状态信息,可以看到数据是否合并成功
print env_3.stat()
def main():
fr = open('lmdb.txt')
# lmdb1的目录
lmdb1 = fr.readline().strip()
# lmdb2的目录
lmdb2 = fr.readline().strip()
# result lmdb的目录
result_lmdb = fr.readline().strip()
fr.close()
merge_lmdb(lmdb1, lmdb2, result_lmdb)
if __name__ == '__main__':
main()
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/187944.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...