我正在使用微调网络进行回归,在 caffe flickr 样式示例的行中,我已将最后一层中的 num 输出更改为 1,但是在使用 matlab 包装函数对图像进行测试时matcaffe_demo()
。它返回对应于单个图像的 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()
,它会给出 10 个输出标签而不是 1。任何人都可以帮助我了解这里发生了什么。提前致谢。