Caffe
include
caffe
util
insert_splits.hpp
1
#ifndef _CAFFE_UTIL_INSERT_SPLITS_HPP_
2
#define _CAFFE_UTIL_INSERT_SPLITS_HPP_
3
4
#include <string>
5
6
#include "caffe/proto/caffe.pb.h"
7
8
namespace
caffe
{
9
10
// Copy NetParameters with SplitLayers added to replace any shared bottom
11
// blobs with unique bottom blobs provided by the SplitLayer.
12
void
InsertSplits(
const
NetParameter& param, NetParameter* param_split);
13
14
void
ConfigureSplitLayer(
const
string
& layer_name,
const
string
& blob_name,
15
const
int
blob_idx,
const
int
split_count,
const
float
loss_weight,
16
LayerParameter* split_layer_param);
17
18
string
SplitLayerName(
const
string
& layer_name,
const
string
& blob_name,
19
const
int
blob_idx);
20
21
string
SplitBlobName(
const
string
& layer_name,
const
string
& blob_name,
22
const
int
blob_idx,
const
int
split_idx);
23
24
}
// namespace caffe
25
26
#endif // CAFFE_UTIL_INSERT_SPLITS_HPP_
caffe
A layer factory that allows one to register layers. During runtime, registered layers can be called b...
Definition:
blob.hpp:14
Generated on Wed May 13 2020 17:45:52 for Caffe by
1.8.17