0

私は微調整されたネットを使用して回帰を行っています.caffe flickrスタイルの例の行で、最後のレイヤーのnum出力を1に変更しましたが、matlabラッパー関数を使用して画像でテストする場合matcaffe_demo(). 1 つの画像に対応する 10 個の出力を返しますが、1 つだけを返す必要があります。

これが私の最後のレイヤーです'deploy.prototxt'

layer {
  name: "my-fc8"
  type: "InnerProduct"
  bottom: "fc7"
  top: "my-fc8"
  # lr_mult is set to higher than for other layers, because this layer is starting from random while the others are already trained
  param {
    lr_mult: 10
    decay_mult: 1
  }      
  param {
    lr_mult: 20
    decay_mult: 0
  }
  inner_product_param {
    num_output: 1
    weight_filler {
    type: "gaussian"
    std: 0.01
  }
  bias_filler {
    type: "constant"
    value: 0
  }
}

ご覧のとおり、num_outputsは 1 です。問題は、 を使用して画像で微調整されたネットをテストすると、matcaffe_demo()1 ではなく 10 の出力ラベルが表示されることです。前もって感謝します。

4

0 に答える 0