由于某些原因,我必须使用老式的“ * .prototxt”文件格式,因此在我的prototxt开头,我必须编写如下内容:
layer {
name: 'input-data'
type: 'Python'
top: 'data'
top: 'gt_boxes'
python_param {
module: 'roi_data_layer.layer'
layer: 'RoIDataLayer'
param_str: "'num_classes': 2"
}
include {
phase: TRAIN
}
}
input: "data"
input_shape {
dim: 1
dim: 3
dim: 224
dim: 224
}
现在,我想使输入层仅适用于TEST阶段,否则将引发“由多个来源生成的caffe顶部blob'数据'”。
我无法为图层设置“ input_param”属性,因为我必须使用过时的分叉版本的Caffe。那怎么办呢?还是有什么方法可以解决此问题?
答案 0 :(得分:0)
layer {
name: 'input-test'
type: 'Input'
top: 'data'
top: 'gt_boxes'
include {
phase: TEST
}
}
我不确定形状规格