diff options
author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2015-10-16 21:11:32 -0700 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2016-02-25 13:32:44 -0800 |
commit | 51f79a837ddea002746d86d69e342a44f099654f (patch) | |
tree | 38e3ac68e30ea5c5dc2bdceb480528591e2935a7 /include | |
parent | bddd04b32c297035b38abbcb41a452bf7167ba17 (diff) | |
download | caffeonacl-51f79a837ddea002746d86d69e342a44f099654f.tar.gz caffeonacl-51f79a837ddea002746d86d69e342a44f099654f.tar.bz2 caffeonacl-51f79a837ddea002746d86d69e342a44f099654f.zip |
drop Net inputs + Forward with bottoms
Drop special cases for `input` fields, the `Net` input members,
and the `Net` interface for Forward with bottoms along with
Forward() / ForwardPrefilled() distinction.
Diffstat (limited to 'include')
-rw-r--r-- | include/caffe/net.hpp | 34 |
1 files changed, 7 insertions, 27 deletions
diff --git a/include/caffe/net.hpp b/include/caffe/net.hpp index 543133e2..43e0a845 100644 --- a/include/caffe/net.hpp +++ b/include/caffe/net.hpp @@ -32,11 +32,10 @@ class Net { void Init(const NetParameter& param); /** - * @brief Run Forward with the input Blob%s already fed separately. + * @brief Run Forward and return the result. * - * You can get the input blobs using input_blobs(). */ - const vector<Blob<Dtype>*>& ForwardPrefilled(Dtype* loss = NULL); + const vector<Blob<Dtype>*>& Forward(Dtype* loss = NULL); /** * The From and To variants of Forward and Backward operate on the @@ -49,14 +48,6 @@ class Net { Dtype ForwardFromTo(int start, int end); Dtype ForwardFrom(int start); Dtype ForwardTo(int end); - /// @brief Run forward using a set of bottom blobs, and return the result. - const vector<Blob<Dtype>*>& Forward(const vector<Blob<Dtype>* > & bottom, - Dtype* loss = NULL); - /** - * @brief Run forward using a serialized BlobProtoVector and return the - * result as a serialized BlobProtoVector - */ - string Forward(const string& input_blob_protos, Dtype* loss = NULL); /** * @brief Zeroes out the diffs of all net parameters. @@ -82,9 +73,9 @@ class Net { */ void Reshape(); - Dtype ForwardBackward(const vector<Blob<Dtype>* > & bottom) { + Dtype ForwardBackward() { Dtype loss; - Forward(bottom, &loss); + Forward(&loss); Backward(); return loss; } @@ -194,18 +185,11 @@ class Net { inline const vector<string>& param_display_names() const { return param_display_names_; } - /// @brief Input and output blob numbers - inline int num_inputs() const { return net_input_blobs_.size(); } + /// @brief output blob number inline int num_outputs() const { return net_output_blobs_.size(); } - inline const vector<Blob<Dtype>*>& input_blobs() const { - return net_input_blobs_; - } inline const vector<Blob<Dtype>*>& output_blobs() const { return net_output_blobs_; } - inline const vector<int>& input_blob_indices() const { - return net_input_blob_indices_; - } inline const vector<int>& output_blob_indices() const { return net_output_blob_indices_; } @@ -229,7 +213,7 @@ class Net { protected: // Helpers for Init. - /// @brief Append a new input or top blob to the net. + /// @brief Append a new top blob to the net. void AppendTop(const NetParameter& param, const int layer_id, const int top_id, set<string>* available_blobs, map<string, int>* blob_name_to_idx); @@ -241,8 +225,6 @@ class Net { void AppendParam(const NetParameter& param, const int layer_id, const int param_id); - /// @brief Helper for displaying debug info in Forward about input Blobs. - void InputDebugInfo(const int layer_id); /// @brief Helper for displaying debug info in Forward. void ForwardDebugInfo(const int layer_id); /// @brief Helper for displaying debug info in Backward. @@ -281,10 +263,8 @@ class Net { vector<string> param_display_names_; vector<pair<int, int> > param_layer_indices_; map<string, int> param_names_index_; - /// blob indices for the input and the output of the net - vector<int> net_input_blob_indices_; + /// blob indices for the output of the net vector<int> net_output_blob_indices_; - vector<Blob<Dtype>*> net_input_blobs_; vector<Blob<Dtype>*> net_output_blobs_; /// The parameters in the network. vector<shared_ptr<Blob<Dtype> > > params_; |