1 #ifndef CAFFE_DATA_LAYER_HPP_ 2 #define CAFFE_DATA_LAYER_HPP_ 6 #include "caffe/blob.hpp" 7 #include "caffe/data_reader.hpp" 8 #include "caffe/data_transformer.hpp" 9 #include "caffe/internal_thread.hpp" 10 #include "caffe/layer.hpp" 11 #include "caffe/layers/base_data_layer.hpp" 12 #include "caffe/proto/caffe.pb.h" 13 #include "caffe/util/db.hpp" 17 template <
typename Dtype>
20 explicit DataLayer(
const LayerParameter& param);
22 virtual void DataLayerSetUp(
const vector<
Blob<Dtype>*>& bottom,
26 virtual inline const char*
type()
const {
return "Data"; }
39 #endif // CAFFE_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
virtual bool ShareInParallel() const
Whether a layer should be shared by multiple nets during data parallelism. By default, all layers except for data layers should not be shared. data layers should be shared to ensure each worker solver access data sequentially during data parallelism.
Definition: data_layer.hpp:25
Definition: data_layer.hpp:18
virtual int ExactNumBottomBlobs() const
Returns the exact number of bottom blobs required by the layer, or -1 if no exact number is required...
Definition: data_layer.hpp:27
virtual const char * type() const
Returns the layer type.
Definition: data_layer.hpp:26
Reads data from a source to queues available to data layers. A single reading thread is created per s...
Definition: data_reader.hpp:23
virtual int MaxTopBlobs() const
Returns the maximum number of top blobs required by the layer, or -1 if no maximum number is required...
Definition: data_layer.hpp:29
virtual int MinTopBlobs() const
Returns the minimum number of top blobs required by the layer, or -1 if no minimum number is required...
Definition: data_layer.hpp:28
A wrapper around SyncedMemory holders serving as the basic computational unit through which Layers...
Definition: blob.hpp:24