深度学习21天实战caffe学习笔记《8:Caffe I/O模块》

运行caffe例程前,详见原始数据转换为LMDB格式,训练网络时需要由数据读取层(DataLayer)不断从LMDB读取数据,送入后续卷积,下采样等层。

1、数据读取层DataLayer:

DataLayer是Layer的派生类,可以读取LMDB、LEVELDB,还可以从原始图像直接读取ImageDataLayer;

(1)数据结构描述

(2)数据读取层实现

声明:include/caffe/data_layers.hpp

实现:src/caffe/layers/base_data_layer.cpp

2、数据变换层Data Transformer:

Data Transformer主要提供对原始输入图像的预处理方法,包括随机切块、随机镜像、幅度缩放、去均值、灰度/色度变换等

(1)数据结构描述

(2)数据读取层实现

声明:include/caffe/data_transformer.hpp

实现:src/caffe/layers/data_transformer.cpp


你可能感兴趣的:(Caffe)