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

Posted 2023-03-12

【中文标题】如何使老式输入层仅用于测试阶段？

【英文标题】：How to make old-fashioned input layers used only for test phase?

【发布时间】：2019-05-07 09:34:05

【问题描述】：

由于某些原因，我不得不使用老式的“*.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 Top blob 'data'”。

我无法为图层设置“input_param”属性，因为我必须使用过时的分叉版本的 Caffe。那么它怎么能做到呢？或者有什么办法可以解决这个问题？

【参考方案1】：

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

    include 
        phase: TEST
    

虽然我不确定形状规格