|
| Net (const NetParameter ¶m) |
|
| Net (const string ¶m_file, Phase phase, const int level=0, const vector< string > *stages=NULL) |
|
void | Init (const NetParameter ¶m) |
| Initialize a network with a NetParameter.
|
|
const vector< Blob< Dtype > * > & | Forward (Dtype *loss=NULL) |
| Run Forward and return the result.
|
|
const vector< Blob< Dtype > * > & | ForwardPrefilled (Dtype *loss=NULL) |
| DEPRECATED; use Forward() instead.
|
|
Dtype | ForwardFromTo (int start, int end) |
|
Dtype | ForwardFrom (int start) |
|
Dtype | ForwardTo (int end) |
|
const vector< Blob< Dtype > * > & | Forward (const vector< Blob< Dtype > * > &bottom, Dtype *loss=NULL) |
| DEPRECATED; set input blobs then use Forward() instead.
|
|
void | ClearParamDiffs () |
| Zeroes out the diffs of all net parameters. Should be run before Backward.
|
|
void | Backward () |
|
void | BackwardFromTo (int start, int end) |
|
void | BackwardFrom (int start) |
|
void | BackwardTo (int end) |
|
void | Reshape () |
| Reshape all layers from bottom to top. More...
|
|
Dtype | ForwardBackward () |
|
void | Update () |
| Updates the network weights based on the diff values computed.
|
|
void | ShareWeights () |
| Shares weight data of owner blobs with shared blobs. More...
|
|
void | ShareTrainedLayersWith (const Net *other) |
| For an already initialized net, implicitly copies (i.e., using no additional memory) the pre-trained layers from another Net.
|
|
void | CopyTrainedLayersFrom (const NetParameter ¶m) |
| For an already initialized net, copies the pre-trained layers from another Net.
|
|
void | CopyTrainedLayersFrom (const string trained_filename) |
|
void | CopyTrainedLayersFromBinaryProto (const string trained_filename) |
|
void | CopyTrainedLayersFromHDF5 (const string trained_filename) |
|
void | ToProto (NetParameter *param, bool write_diff=false) const |
| Writes the net to a proto.
|
|
void | ToHDF5 (const string &filename, bool write_diff=false) const |
| Writes the net to an HDF5 file.
|
|
const string & | name () const |
| returns the network name.
|
|
const vector< string > & | layer_names () const |
| returns the layer names
|
|
const vector< string > & | blob_names () const |
| returns the blob names
|
|
const vector< shared_ptr< Blob< Dtype > > > & | blobs () const |
| returns the blobs
|
|
const vector< shared_ptr< Layer< Dtype > > > & | layers () const |
| returns the layers
|
|
Phase | phase () const |
| returns the phase: TRAIN or TEST
|
|
const vector< vector< Blob< Dtype > * > > & | bottom_vecs () const |
| returns the bottom vecs for each layer – usually you won't need this unless you do per-layer checks such as gradients.
|
|
const vector< vector< Blob< Dtype > * > > & | top_vecs () const |
| returns the top vecs for each layer – usually you won't need this unless you do per-layer checks such as gradients.
|
|
const vector< int > & | top_ids (int i) const |
| returns the ids of the top blobs of layer i
|
|
const vector< int > & | bottom_ids (int i) const |
| returns the ids of the bottom blobs of layer i
|
|
const vector< vector< bool > > & | bottom_need_backward () const |
|
const vector< Dtype > & | blob_loss_weights () const |
|
const vector< bool > & | layer_need_backward () const |
|
const vector< shared_ptr< Blob< Dtype > > > & | params () const |
| returns the parameters
|
|
const vector< Blob< Dtype > * > & | learnable_params () const |
|
const vector< float > & | params_lr () const |
| returns the learnable parameter learning rate multipliers
|
|
const vector< bool > & | has_params_lr () const |
|
const vector< float > & | params_weight_decay () const |
| returns the learnable parameter decay multipliers
|
|
const vector< bool > & | has_params_decay () const |
|
const map< string, int > & | param_names_index () const |
|
const vector< int > & | param_owners () const |
|
const vector< string > & | param_display_names () const |
|
int | num_inputs () const |
| Input and output blob numbers.
|
|
int | num_outputs () const |
|
const vector< Blob< Dtype > * > & | input_blobs () const |
|
const vector< Blob< Dtype > * > & | output_blobs () const |
|
const vector< int > & | input_blob_indices () const |
|
const vector< int > & | output_blob_indices () const |
|
bool | has_blob (const string &blob_name) const |
|
const shared_ptr< Blob< Dtype > > | blob_by_name (const string &blob_name) const |
|
bool | has_layer (const string &layer_name) const |
|
const shared_ptr< Layer< Dtype > > | layer_by_name (const string &layer_name) const |
|
void | set_debug_info (const bool value) |
|
const vector< Callback * > & | before_forward () const |
|
void | add_before_forward (Callback *value) |
|
const vector< Callback * > & | after_forward () const |
|
void | add_after_forward (Callback *value) |
|
const vector< Callback * > & | before_backward () const |
|
void | add_before_backward (Callback *value) |
|
const vector< Callback * > & | after_backward () const |
|
void | add_after_backward (Callback *value) |
|
|
void | AppendTop (const NetParameter ¶m, const int layer_id, const int top_id, set< string > *available_blobs, map< string, int > *blob_name_to_idx) |
| Append a new top blob to the net.
|
|
int | AppendBottom (const NetParameter ¶m, const int layer_id, const int bottom_id, set< string > *available_blobs, map< string, int > *blob_name_to_idx) |
| Append a new bottom blob to the net.
|
|
void | AppendParam (const NetParameter ¶m, const int layer_id, const int param_id) |
| Append a new parameter blob to the net.
|
|
void | ForwardDebugInfo (const int layer_id) |
| Helper for displaying debug info in Forward.
|
|
void | BackwardDebugInfo (const int layer_id) |
| Helper for displaying debug info in Backward.
|
|
void | UpdateDebugInfo (const int param_id) |
| Helper for displaying debug info in Update.
|
|
| DISABLE_COPY_AND_ASSIGN (Net) |
|
|
string | name_ |
| The network name.
|
|
Phase | phase_ |
| The phase: TRAIN or TEST.
|
|
vector< shared_ptr< Layer< Dtype > > > | layers_ |
| Individual layers in the net.
|
|
vector< string > | layer_names_ |
|
map< string, int > | layer_names_index_ |
|
vector< bool > | layer_need_backward_ |
|
vector< shared_ptr< Blob< Dtype > > > | blobs_ |
| the blobs storing intermediate results between the layer.
|
|
vector< string > | blob_names_ |
|
map< string, int > | blob_names_index_ |
|
vector< bool > | blob_need_backward_ |
|
vector< vector< Blob< Dtype > * > > | bottom_vecs_ |
|
vector< vector< int > > | bottom_id_vecs_ |
|
vector< vector< bool > > | bottom_need_backward_ |
|
vector< vector< Blob< Dtype > * > > | top_vecs_ |
| top_vecs stores the vectors containing the output for each layer
|
|
vector< vector< int > > | top_id_vecs_ |
|
vector< Dtype > | blob_loss_weights_ |
|
vector< vector< int > > | param_id_vecs_ |
|
vector< int > | param_owners_ |
|
vector< string > | param_display_names_ |
|
vector< pair< int, int > > | param_layer_indices_ |
|
map< string, int > | param_names_index_ |
|
vector< int > | net_input_blob_indices_ |
| blob indices for the input and the output of the net
|
|
vector< int > | net_output_blob_indices_ |
|
vector< Blob< Dtype > * > | net_input_blobs_ |
|
vector< Blob< Dtype > * > | net_output_blobs_ |
|
vector< shared_ptr< Blob< Dtype > > > | params_ |
| The parameters in the network.
|
|
vector< Blob< Dtype > * > | learnable_params_ |
|
vector< int > | learnable_param_ids_ |
|
vector< float > | params_lr_ |
| the learning rate multipliers for learnable_params_
|
|
vector< bool > | has_params_lr_ |
|
vector< float > | params_weight_decay_ |
| the weight decay multipliers for learnable_params_
|
|
vector< bool > | has_params_decay_ |
|
size_t | memory_used_ |
| The bytes of memory used by this net.
|
|
bool | debug_info_ |
| Whether to compute and display debug info for the net.
|
|
vector< Callback * > | before_forward_ |
|
vector< Callback * > | after_forward_ |
|
vector< Callback * > | before_backward_ |
|
vector< Callback * > | after_backward_ |
|
template<typename Dtype>
class caffe::Net< Dtype >
Connects Layers together into a directed acyclic graph (DAG) specified by a NetParameter.
TODO(dox): more thorough description.