| 21 | |
| 22 | template<typename Dtype> |
| 23 | DataTransformer<Dtype>::DataTransformer(const TransformationParameter& param, |
| 24 | Phase phase) |
| 25 | : param_(param), phase_(phase) { |
| 26 | // check if we want to use mean_file |
| 27 | if (param_.has_mean_file()) { |
| 28 | CHECK_EQ(param_.mean_value_size(), 0) << |
| 29 | "Cannot specify mean_file and mean_value at the same time"; |
| 30 | const string& mean_file = param.mean_file(); |
| 31 | if (Caffe::root_solver()) { |
| 32 | LOG(INFO) << "Loading mean file from: " << mean_file; |
| 33 | } |
| 34 | BlobProto blob_proto; |
| 35 | ReadProtoFromBinaryFileOrDie(mean_file.c_str(), &blob_proto); |
| 36 | data_mean_.FromProto(blob_proto); |
| 37 | } |
| 38 | // check if we want to use mean_value |
| 39 | if (param_.mean_value_size() > 0) { |
| 40 | CHECK(param_.has_mean_file() == false) << |
| 41 | "Cannot specify mean_file and mean_value at the same time"; |
| 42 | for (int c = 0; c < param_.mean_value_size(); ++c) { |
| 43 | mean_values_.push_back(param_.mean_value(c)); |
| 44 | } |
| 45 | } |
| 46 | |
| 47 | // CPM extra code: |
| 48 | LOG(INFO) << "DataTransformer constructor done."; |
| 49 | np_in_lmdb = param_.np_in_lmdb(); |
| 50 | LOG(INFO) << "np_in_lmdb" << np_in_lmdb; |
| 51 | np = param_.num_parts(); |
| 52 | is_table_set = false; |
| 53 | // CPM end extra code |
| 54 | } |
| 55 | |
| 56 | template<typename Dtype> |
| 57 | void DataTransformer<Dtype>::Transform(const Datum& datum, |
nothing calls this directly
no test coverage detected