Question

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

Answer 1

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

    include {
        phase: TEST
    }
}

我不确定形状规格

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

1 个答案: