我正在使用caffe的一个旧分支。现在我需要通过切分输入层来修改prototxt文件。
我知道在新语法中,它看起来是这样的:
layer { name: "slice" type: "Slice" bottom: "labelAndMask" ## 图层形状为 N x 5 x Height x Width 的示例 top: "label" top: "mask" slice_param { axis: 1 slice_point: 1 }}
在旧的prototxt格式中,相应的写法是什么?另外,我在caffe源码中应该去哪里查找这些信息?
回答:
你应该查看$CAFFE_ROOT/src/caffe/proto/caffe.proto
文件的底部,你会看到V1LayerParameter
的定义。
对于旧语法中的切分层:
layers { type: SLICE # 这不是字符串,而是一个枚举值 name: "slice" bottom: "labelAndMask" ## 图层形状为 N x 5 x Height x Width 的示例 top: "label" top: "mask" slice_param { axis: 1 slice_point: 1 }}