1 #ifndef CAFFE_DATA_TRANSFORMER_HPP
2 #define CAFFE_DATA_TRANSFORMER_HPP
6 #include "caffe/blob.hpp"
7 #include "caffe/common.hpp"
8 #include "caffe/proto/caffe.pb.h"
16 template <
typename Dtype>
19 explicit DataTransformer(
const TransformationParameter& param, Phase phase);
50 void Transform(
const vector<Datum> & datum_vector,
64 void Transform(
const vector<cv::Mat> & mat_vector,
139 virtual int Rand(
int n);
141 void Transform(
const Datum& datum, Dtype* transformed_data);
143 TransformationParameter param_;
146 shared_ptr<Caffe::RNG> rng_;
149 vector<Dtype> mean_values_;
154 #endif // CAFFE_DATA_TRANSFORMER_HPP_
A wrapper around SyncedMemory holders serving as the basic computational unit through which Layers,...
Definition: blob.hpp:24
A layer factory that allows one to register layers. During runtime, registered layers can be called b...
Definition: blob.hpp:14