caffe 将三通道或四通道图片转换为lmdb格式,将标签(单通道灰度图)转换为lmdb格式
2016-11-28 11:25
471 查看
import numpy as np import lmdb from PIL import Image import sys # import caffe module caffe_root = '/home/user/SegNet/caffe-segnet/' sys.path.insert(0, caffe_root + 'python') import caffe # # read file train_file = open('/home/user/train.txt') inputs_data_train = train_file.readlines() train_file.close() print("Creating Training Data LMDB File ..... ") in_db = lmdb.open('/home/user/Val_Data_lmdb', map_size=int(1e12)) with in_db.begin(write=True) as in_txn: for in_idx, in_ in enumerate(inputs_data_train): # print in_idx in_ = in_.strip() im = np.array(Image.open(in_)) Dtype = im.dtype if im.shape[2]== 3: print('The image has 3 channel') # RGB to BGR im = im[:, :, ::-1] if im.shape[2]==4: im3 = im[:,:,0:3] im3 = np.array(im3) #RGB to BGR im3 = im3[:,:,::-1] im[:, :, 0] = im3[:, :, 0] im[:, :, 1] = im3[:, :, 1] im[:, :, 2] = im3[:, :, 2] im = Image.fromarray(im) im = np.array(im, Dtype) im = im.transpose((2, 0, 1)) im_dat = caffe.io.array_to_datum(im) in_txn.put('{:0>10d}'.format(in_idx), im_dat.SerializeToString()) in_db.close() # read file label_file = open('/home/user/trainannot.txt') inputs_data_label = label_file.readlines() label_file.close() print("Creating Training Label LMDB File ..... ") #map_size:Change the maximum size of the map file in_db1 = lmdb.open('//home/user/Val_Label_Data_lmdb',map_size=int(1e12)) with in_db1.begin(write=True) as in_txn: for in_idx, in_ in enumerate(inputs_data_label): in_ = in_.strip() Dtype = 'uint8' L = np.array(Image.open(in_), Dtype) Limg = Image.fromarray(L) L = np.array(Limg,Dtype) L = L.reshape(L.shape[0],L.shape[1],1) L = L.transpose((2,0,1)) L_dat = caffe.io.array_to_datum(L) in_txn.put('{:0>10d}'.format(in_idx),L_dat.SerializeToString()) in_db1.close() print("Finish creating lmdb file ......")
相关文章推荐
- caffe 将三通道或四通道图片转换为lmdb格式,将标签(单通道灰度图)转换为lmdb格式
- 深度学习caffe应用笔记--如何将图片转换为lmdb格式
- Caffe下将图片转换为lmdb格式
- [转]caffe下为图像加标签,转换为可执行文件.lmdb格式
- 用Python将多标签数据存入caffe使用的HDF5&LMDB格式
- 深度学习caffe平台--制作自己.lmdb格式数据集及分类标签文件
- 【深度学习】【caffe实用工具3】笔记25 Windows下caffe中将图像数据集合转换为DB(LMDB/LEVELDB)文件格式之convert_imageset
- 将图片转换成caffe的lmdb形式并进行实验
- 【caffe-windows】 caffe-master 之图片转换成lmdb or leveldb
- linux caffe 将图片转为lmdb格式
- CAFFE之将图片转化为lmdb格式脚本【直接运行版】(内含生成train.txt和test.txt)
- 将数据转换为caffe可用的lmdb格式(convert_data_lmdb.sh 解析)
- 利用caffe工具将图片转换成lmdb
- 将图片转换成caffe的数据格式
- 深度学习caffe平台--制作自己.lmdb格式数据集及分类标签文件
- caffe中怎么把自己的图片转化成lmdb格式
- caffe神经网络框架的辅助工具(将图片转换为leveldb格式)
- Caffe框架,图像数据转换成LMDB数据格式
- Caffe:如何将图片数据转换成lmdb文件
- caffe与cv2图片格式的相互转换