| 19 | |
| 20 | template <typename Dtype> |
| 21 | void BaseDataLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype>*>& bottom, |
| 22 | const vector<Blob<Dtype>*>& top) { |
| 23 | if (top.size() == 1) { |
| 24 | output_labels_ = false; |
| 25 | } else { |
| 26 | output_labels_ = true; |
| 27 | } |
| 28 | data_transformer_.reset( |
| 29 | new DataTransformer<Dtype>(transform_param_, this->phase_)); |
| 30 | data_transformer_->InitRand(); |
| 31 | // The subclasses should setup the size of bottom and top |
| 32 | DataLayerSetUp(bottom, top); |
| 33 | } |
| 34 | |
| 35 | template <typename Dtype> |
| 36 | BasePrefetchingDataLayer<Dtype>::BasePrefetchingDataLayer( |
nothing calls this directly
no test coverage detected