0

我想微调 VGG19,它运行良好。在训练时,我将平均像素减去

name: "VGG_ILSVRC_19_layer"
layer {
  name: "data"
  type: "Data"
  include {
    phase: TRAIN
  }
 transform_param {
    mean_value: 104
    mean_value: 117
    mean_value: 123
    mirror: false
 }
 data_param {
    source: "examples/VGG_finetune/train_lmdb"
    batch_size: 8
    backend: LMDB
  }
  top: "data"
  top: "label"
}

现在我需要通过我的微调网络进行前馈。如果我看一下,他们 没有在部署文件中指定平均像素值。

问题:
如何在前馈时提供我的网络平均像素值?
我的前馈代码(加载网络)是

net = caffe.Classifier(model_prototxt, model_trained,
                           mean=[104,117,123],
                           channel_swap=(2,1,0),
                           raw_scale=255,
                           image_dims=(224, 224))

现在我不确定是否mean=(104,117,123)工作,因为当我查看源代码时,classifier我通过代码中的注释了解到它只需要ndarray否则它会给出错误。
如何从输入图像中减去平均像素?

4

1 回答 1

3

你可以简单地指定你meanndarray

net = caffe.Classifier(model_prototxt, model_trained,
                       mean=NP.array( [104, 117, 123], dtype='f4'),
                       channel_swap=(2,1,0),
                       raw_scale=255,
                       image_dims=(224, 224))
于 2016-07-14T12:33:56.527 回答