私は微調整されたネットを使用して回帰を行っています.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 の出力ラベルが表示されることです。前もって感謝します。