如何在deploy.prototxt中指定tranform_param

时间:2016-10-18 17:25:36

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

在caffe中的train_val原型文件文件中构建deploy.prototxt时,如何在部署文件中指定转换参数。 例如,我在train_val.prototxt中有以下数据层:

    layer 
{
        name: "data"
        type: "Data"
        include {
          phase: TRAIN
       }
      data_param 
{
        source: "/path/to/train_lmdb"
        batch_size: 32
        backend: LMDB
      }
      transform_param {
        crop_size: 224
        mean_value: 103.939
        mean_value: 116.779
        mean_value: 123.68
        mirror: true
      }
    }

我知道我必须从deploy.prototxt中删除此图层,但是如何指定这些平均值呢?还是裁剪尺寸?

在我的情况下,部署文件看起来像

name: "VGG_ILSVRC_19_layers"
input: "data"
input_dim: 10
input_dim: 3
input_dim: 224
input_dim: 224

但是我在哪里指定transform_param?

1 个答案:

答案 0 :(得分:0)

当您部署网络时,您有责任以与培训期间相同的方式转换输入