caffe 将三通道或四通道图片转换为lmdb格式,将标签(单通道灰度图)转换为lmdb格式
2017-07-01 10:22
645 查看
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 ......")
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:如何将图片数据转换成lmdb文件
- linux caffe 将图片转为lmdb格式
- 将数据转换为caffe可用的lmdb格式(convert_data_lmdb.sh 解析)
- Caffe下将mnist手写图片数据转化成lmdb格式
- 将图片转换成caffe的数据格式
- 深度学习caffe平台--制作自己.lmdb格式数据集及分类标签文件
- caffe中怎么把自己的图片转化成lmdb格式
- caffe神经网络框架的辅助工具(将图片转换为leveldb格式)
- Caffe框架,图像数据转换成LMDB数据格式
- 利用caffe工具将图片转换成lmdb
- caffe与cv2图片格式的相互转换
- caffe学习之conver_imageset.bin的使用方法,caffe下图片转lmdb格式类型数据总结