python读取LMDB中图像的方法
本文实例为大家分享了python读取LMDB中的图像具体代码,供大家参考,具体内容如下
图像数据写入LMDB之后最好再按照写入的逻辑反向解析写入的图像,如果图像能够被还原则证明写入方式是没有问题的。
fromPILimportImage defread_from_lmdb(lmdb_path,img_save_to): try: lmdb_env=lmdb.open(lmdb_path,map_size=3221225472) lmdb_txn=lmdb_env.begin() lmdb_cursor=lmdb_txn.cursor() datum=caffe.Datum() datum_index=0 forkey,valueinlmdb_cursor: datum.ParseFromString(value) label=datum.label data=datum.data channel=datum.channels print('Datumchannels:%d'%datum.channels) print('Datumwidth:%d'%datum.width) print('Datumheight:%d'%datum.height) print('Datumdatalength:%d'%len(datum.data)) print('Datumlabel:%d'%datum.label) size=datum.width*datum.height pixles1=datum.data[0:size] pixles2=datum.data[size:2*size] pixles3=datum.data[2*size:3*size] #Extractimagesofdifferentchannel image1=Image.frombytes('L',(datum.width,datum.height),pixles1) image2=Image.frombytes('L',(datum.width,datum.height),pixles2) image3=Image.frombytes('L',(datum.width,datum.height),pixles3) #注意三通道的顺序,如果LMDB中图像是按照BGR存储的则需要按照:image3,image2,image1的顺序合并为RGB图像。PIL中图像是按照RGB的顺序存储的 image4=Image.merge("RGB",(image3,image2,image1)) image4.save(img_save_to+str(key)+".jpg") datum_index+=1 print"extracted" finally: lmdb_env.close()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。