caffe python lmdb
来源:互联网 发布:群晖网络存储 编辑:程序博客网 时间:2024/05/21 06:25
import numpy as np
import os
import matplotlib.pyplot as plt
import lmdb
from PIL import Image
import random
import sys
caffe_root = '/home/tsq/Documents/project/mcnn/'
sys.path.insert(0, caffe_root + 'python')
import caffe
train_file = open('test_b.txt')
inputs_data_train = train_file.readlines()
train_file.close()
print("Creating Training Data LMDB File ..... ")
in_db = lmdb.open('Train_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()
in_='test_b/'+in_
im = np.array(Image.open(in_))
Dtype = im.dtype
if len(im.shape) == 2:
print('here')
(row, col) = im.shape
im3 = np.zeros([row, col, 3], Dtype)
for i in range(3):
im3 [:, :, i] = im
im = im3
print('here')
im = im[:,:,::-1]
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()
import scipy.io as sio
label_file = open('test_b_gt.txt')
inputs_data_label = label_file.readlines()
label_file.close()
print("Creating Training Label LMDB File ..... ")
in_db1 = lmdb.open('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()
in_='gt2/'+in_
Dtype = 'double'
L = sio.loadmat(in_)
L = np.array(L['d_map'], Dtype)
L=np.expand_dims(L,axis=0)
L_dat = caffe.io.array_to_datum(L,0)
in_txn.put('{:0>10d}'.format(in_idx),L_dat.SerializeToString())
in_db1.close()
print("Finish creating lmdb file ......")
import os
import matplotlib.pyplot as plt
import lmdb
from PIL import Image
import random
import sys
caffe_root = '/home/tsq/Documents/project/mcnn/'
sys.path.insert(0, caffe_root + 'python')
import caffe
train_file = open('test_b.txt')
inputs_data_train = train_file.readlines()
train_file.close()
print("Creating Training Data LMDB File ..... ")
in_db = lmdb.open('Train_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()
in_='test_b/'+in_
im = np.array(Image.open(in_))
Dtype = im.dtype
if len(im.shape) == 2:
print('here')
(row, col) = im.shape
im3 = np.zeros([row, col, 3], Dtype)
for i in range(3):
im3 [:, :, i] = im
im = im3
print('here')
im = im[:,:,::-1]
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()
import scipy.io as sio
label_file = open('test_b_gt.txt')
inputs_data_label = label_file.readlines()
label_file.close()
print("Creating Training Label LMDB File ..... ")
in_db1 = lmdb.open('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()
in_='gt2/'+in_
Dtype = 'double'
L = sio.loadmat(in_)
L = np.array(L['d_map'], Dtype)
L=np.expand_dims(L,axis=0)
L_dat = caffe.io.array_to_datum(L,0)
in_txn.put('{:0>10d}'.format(in_idx),L_dat.SerializeToString())
in_db1.close()
print("Finish creating lmdb file ......")
0 0
- caffe python lmdb
- caffe python lmdb读写
- caffe中用python读写lmdb
- Python lmdb
- python-lmdb
- caffe LMDB 库学习
- caffe的lmdb
- Caffe convert to lmdb
- caffe 数据库LMDB的读写
- Caffe: LMDB 及其数据转换
- Caffe中LMDB的使用
- caffe 预处理,图片转为LMDB
- caffe 制作lmdb数据集
- 用Python将多标签数据存入caffe使用的HDF5&LMDB格式
- Caffe: MNIST 数据集格式转换、用 python 读写 LMDB 数据库
- 【用Python学习Caffe】3. 图像训练测试数据集LMDB的生成
- caffe 利用python API 将数据转成lmdb格式的数据
- caffe用python加载数据,包含各类数据类型(LMDB,image,HDF5等共五种全部格式))--caffe学习(3)
- 电容,电压 Q, I, T 关系
- 【Java】try、catch、finally的执行顺序与返回值问题
- 常见控件简单总结
- mac使用ssh-keygen和ssh-copy-id三步实现SSH无密码登录
- 海量数据相似度计算之simhash和海明距离
- caffe python lmdb
- Hadoop之——Hadoop2.5.2 HA高可靠性集群搭建(Hadoop+Zookeeper)
- BSOJ:3362 聪明的农民 贪心
- 感知机
- Android 消息处理机制原理
- Java中File的常用操作及方法
- caffe-custom network input/image_data_layer/hdf5_data_layer
- Ajax访问远程WebService 配置文件、编码工作注意
- Linux_VMware虚拟机下为什么不能识别USB设备