Как сделать старомодные входные слои, используемые только для фазы тестирования?

Question

По некоторым причинам, я должен использовать старомодный формат файла "* .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, созданные несколькими источниками».

Я не могу установить атрибут «input_param» для слоя, так как у меня естьработать над устаревшей раздвоенной версией Caffe.Так как же это сделать?Или есть ли способ обойти эту проблему?

Answer 1

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

    include {
        phase: TEST
    }
}

Я не уверен насчет спецификации формы, хотя