如何使老式的输入层仅用于测试阶段?

时间:2018-12-05 18:34:19

标签: machine-learning neural-network deep-learning conv-neural-network caffe

由于某些原因,我必须使用老式的“ * .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。那怎么办呢?还是有什么方法可以解决此问题?

1 个答案:

答案 0 :(得分:0)

layer {
    name: 'input-test'
    type: 'Input'
    top: 'data'
    top: 'gt_boxes'

    include {
        phase: TEST
    }
}

我不确定形状规格