caffe Python API 之 数据输入层(Data,ImageData,HDF5Data)
2024-08-28 08:58:03
import sys
sys.path.append('/projects/caffe-ssd/python')
import caffe
4 net = caffe.NetSpec()
一、ImageData Layer
net.data ,net.label = caffe.layers.ImageData(
name="InputData"
source="train.txt",
batch_size=32,
new_width=48,
new_height=48,
ntop=2,
is_color=True,
shuffle=True,
root_folder='/',
transform_param=dict(crop_size=40,mirror=True))
print str(net.to_proto()) 输出:
layer {
name: "InputData"
type: "ImageData"
top: "data"
top: "label"
transform_param {
mirror: true
crop_size: 40
}
image_data_param {
source: "train.txt"
batch_size: 32
shuffle: true
new_height: 48
new_width: 48
is_color: true
root_folder: "/"
}
}
二、Data Layer (lmdb/leveldb)
net.data, net.label = caffe.layers.Data(
name="InputData",
source="train_lmdb",
backend = caffe.params.Data.LMDB,
batch_size=32,
ntop=2,
include=dict(phase=caffe.TRAIN)
transform_param=dict(
crop_size=227,
mean_value=[104, 117, 123],
mirror=True
)
) 输出:
layer {
name: "InputData"
type: "Data"
top: "data"
top: "label"
transform_param {
mirror: true
crop_size: 227
mean_value: 104
mean_value: 117
mean_value: 123
}
data_param {
source: "train_lmdb"
batch_size: 32
backend: LMDB
}
}
三、HDF5Data Layer
net.data, net.label = caffe.layers.HDF5Data(
name="InputData",
source='./training_data_paths.txt',
batch_size=64,
include=dict(phase=caffe.TRAIN),
ntop=2
) 输出:
layer {
name: "InputData"
type: "HDF5Data"
top: "data"
top: "label"
include {
phase: TRAIN
}
hdf5_data_param {
source: "./training_data_paths.txt"
batch_size: 64
}
} 另有:
image = L.HDF5Data(
hdf5_data_param={ 'source': './training_data_paths.txt', 'batch_size': 64 },
include={'phase': caffe.TRAIN }
)
最新文章
- git-credential-winstore.exe";: No such file or directory
- 重写Object类中的equals方法
- Thinkphp---------Call to a member function free_result() on a non-object
- [leetcode] Contains Duplicate II
- ASP.NET MVC4 学习系统四(视图)
- HTTP协议的状态码
- HDU 5688 Problem D
- oralce表空间自增长占满磁盘
- Quartz1.8.5例子(八)
- php 导出 Excel 报错 exception 'PHPExcel_Calculation_Exception' with message
- 复制粘贴之插件(jquery.zclip.min.js)需要安装flash
- mysql__存储过程
- gradle构建项目失败:Unzipping /home/.gradle/wrapper/dists/gradle-3.3-all/55gk2rcmfc6p2dg9u9ohc3hw9/gradle-3.3-all.zip to /home/.gradle/wrapper/dists/gradle-3.3-all/55gk2rcmfc6p2dg9u9ohc3hw9
- react native 5.54 出ios版本遇到的坑(应该是在xcode10下才会有的吧)记录。。。。。。 据说5.7已经修复了
- django 三种缓存模式的使用及注意点
- [学习]UX 测试 5S 范围
- js调用打印机打印
- 关于订单BOM替换组件不成功的问题
- POJ 1265 pick定理
- Linux系统部署应用ECShop