caffe使用python创建lmdb

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了caffe使用python创建lmdb，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1100字，纯文字阅读大概需要2分钟。

内容图文

import numpy as np 2 import sys 3 from PIL import Image 4 import lmdb 5 import random 6 import os 7 8 sys.path.append(‘../caffe/python/‘) 9 10 import caffe 11 12 if __name__ == ‘__main__‘ : 13 train_list_file = ‘./trainlist.txt‘ 14 train_images_root = ‘./train/images/‘ 15 16 f = open(train_list_file, ‘r‘) 17 trainlist = f.readlines() 18 f.close() 19 20 random.shuffle(trainlist) 21 22 # creating images lmdb 23 in_db = lmdb.open(‘my-train-data-lmdb‘, map_size=int(1e12)) 24 with in_db.begin(write=True) as in_txn : 25 for in_idx, in_ in enumerate(trainlist) : 26 fid = in_.strip() 27 fn = os.path.join(train_images_root, fid) 28 im = np.array(Image.open(fn)) 29 im = im[:,:,::-1] 30 im = im.transpose((2, 0, 1)) 31 im_dat = caffe.io.array_to_datum(im) 32 in_txn.put(‘{:0>10d}‘.format(in_idx), im_dat.SerializeToString()) 33 in_db.close()

 

caffe使用python创建lmdb

标签：

本文系统来源：http://www.cnblogs.com/leemo-o/p/4990021.html

内容总结

以上是互联网集市为您收集整理的caffe使用python创建lmdb全部内容，希望文章能够帮你解决caffe使用python创建lmdb所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---