Net类是Solve类的一个成员,在net.cpp中定义了对Net的所有操作,其中包括:
- Init
- GetLearningRateAndWeightDecay
- ForwardPrefilled
- Backward
- ShareTrainedLayersWith
- CopyTrainedLayersFrom
- ToProto
- Update
- has_blob
- blob_by_name
- has_layer
- layer_by_name
Net(const NetParameter& param)
功能:调用Init函数初始化网络
输入:NetParameter& param
输出:无
Net(const string& param_file)
功能:调用Init函数初始化网络
输入:string& param_file
输出:无
Init(const NetParameter& in_param)
功能:初始化网络
输入:NetParameter& in_param
输出:无
步骤:
<1> 调用InsertSplits()函数从in_param读入新网络到param
<2> 定义name_,blob_name_to_idx,available_blobs,num_layers
<3> param.input_size()返回输入层blob的个数;
param.input(i)表示第i个blob的名字;
param.layers_size()返回网络的层数。
<4> 对每一个输入层的blob:
- 产生一块和当前blob一样大的空间 e.g. imput_dim=[12 55 66 39 20 24 48 64]表示第一个blob的四个维数为 12 55 66 39,第二个为 20 24 48 64 接着blob_pointer指向这块空间
- blob_pointer压到blobs_中
vector<shared_ptr<Blob<Dtype>>> blobs_
- blob_name压到blob_names_中
vector<string> blob_names_
- param.force_backward()压到blob_need_backward_中
vector<bool> blob_need_backward_
- i 压到 net_input_blob_indices_中 net_input_blob_indices_ -> vector
- blob_pointer.get() 压