| | #ifndef CAFFE_DUMMY_DATA_LAYER_HPP_ |
| | #define CAFFE_DUMMY_DATA_LAYER_HPP_ |
| |
|
| | #include <vector> |
| |
|
| | #include "caffe/blob.hpp" |
| | #include "caffe/filler.hpp" |
| | #include "caffe/layer.hpp" |
| | #include "caffe/proto/caffe.pb.h" |
| |
|
| | namespace caffe { |
| |
|
| | |
| | |
| | |
| | |
| | |
| | template <typename Dtype> |
| | class DummyDataLayer : public Layer<Dtype> { |
| | public: |
| | explicit DummyDataLayer(const LayerParameter& param) |
| | : Layer<Dtype>(param) {} |
| | virtual void LayerSetUp(const vector<Blob<Dtype>*>& bottom, |
| | const vector<Blob<Dtype>*>& top); |
| | |
| | virtual void Reshape(const vector<Blob<Dtype>*>& bottom, |
| | const vector<Blob<Dtype>*>& top) {} |
| |
|
| | virtual inline const char* type() const { return "DummyData"; } |
| | virtual inline int ExactNumBottomBlobs() const { return 0; } |
| | virtual inline int MinTopBlobs() const { return 1; } |
| |
|
| | protected: |
| | virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom, |
| | const vector<Blob<Dtype>*>& top); |
| | virtual void Backward_cpu(const vector<Blob<Dtype>*>& top, |
| | const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {} |
| | virtual void Backward_gpu(const vector<Blob<Dtype>*>& top, |
| | const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {} |
| |
|
| | vector<shared_ptr<Filler<Dtype> > > fillers_; |
| | vector<bool> refill_; |
| | }; |
| |
|
| | } |
| |
|
| | #endif |
| |
|