LMDB To Picture by Python

最新推荐文章于 2022-03-10 16:25:40 发布

Sunshine_in_Moon

最新推荐文章于 2022-03-10 16:25:40 发布

阅读量603

点赞数

分类专栏： Python caffe

本文链接：https://blog.csdn.net/Sunshine_in_Moon/article/details/79052989

版权

Python 同时被 2 个专栏收录

52 篇文章 0 订阅

订阅专栏

caffe

40 篇文章 1 订阅

订阅专栏

自己写的一个小代码，稍后会传到GitHub上，这里分享给大家！

# coding: utf-8

import lmdb
import numpy as np
import caffe_pb2
import cv2

dataDir = '***_lmdb'#lmdb文件夹

env = lmdb.open(dataDir,readonly=True,lock=True)
datum = caffe_pb2.Datum()
txn = env.begin()

fr_key = open('key','r')#存放key的文件，根据key去找数据
keylines = fr_key.readline()
fr_key.close()

for k in keylines:
rawDatum = txn.get(k.strip())
datum.ParseFromString(rawDatum)
flatData = np.fromstring(datum.data,dtype=np.uint8)
x = flatData.reshape(datum.channels,datum.height,datum.wdith)
x = x.transpose(1,2,0)
cv2.imshow('pic','x')
cv2.waitKey(0)
cv2.destroyAllwindows()