0

我正在使用微调网络进行回归,在 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。任何人都可以帮助我了解这里发生了什么。提前致谢。

4

0 回答 0