1 #ifndef CAFFE_WINDOW_DATA_LAYER_HPP_ 2 #define CAFFE_WINDOW_DATA_LAYER_HPP_ 8 #include "caffe/blob.hpp" 9 #include "caffe/data_transformer.hpp" 10 #include "caffe/internal_thread.hpp" 11 #include "caffe/layer.hpp" 12 #include "caffe/layers/base_data_layer.hpp" 13 #include "caffe/proto/caffe.pb.h" 23 template <
typename Dtype>
29 virtual void DataLayerSetUp(
const vector<
Blob<Dtype>*>& bottom,
32 virtual inline const char*
type()
const {
return "WindowData"; }
37 virtual unsigned int PrefetchRand();
40 shared_ptr<Caffe::RNG> prefetch_rng_;
41 vector<std::pair<std::string, vector<int> > > image_database_;
42 enum WindowField { IMAGE_INDEX, LABEL, OVERLAP, X1, Y1, X2, Y2, NUM };
43 vector<vector<float> > fg_windows_;
44 vector<vector<float> > bg_windows_;
46 vector<Dtype> mean_values_;
48 bool has_mean_values_;
50 vector<std::pair<std::string, Datum > > image_database_cache_;
55 #endif // CAFFE_WINDOW_DATA_LAYER_HPP_ Definition: base_data_layer.hpp:49
Definition: base_data_layer.hpp:55
A layer factory that allows one to register layers. During runtime, registered layers can be called b...
Definition: blob.hpp:14
Provides data to the Net from windows of images files, specified by a window data file...
Definition: window_data_layer.hpp:24
virtual int ExactNumBottomBlobs() const
Returns the exact number of bottom blobs required by the layer, or -1 if no exact number is required...
Definition: window_data_layer.hpp:33
virtual const char * type() const
Returns the layer type.
Definition: window_data_layer.hpp:32
virtual int ExactNumTopBlobs() const
Returns the exact number of top blobs required by the layer, or -1 if no exact number is required...
Definition: window_data_layer.hpp:34
A wrapper around SyncedMemory holders serving as the basic computational unit through which Layers...
Definition: blob.hpp:24