c++ - 计算 Caffe 中的 top-5 错误率？

Question

我使用 synset 计算k来自 softmax 输出的排序最高预测。

这给了我前 5 个类名。但我想知道如何计算它的百分比。我的意思是前 5% 的错误。

任何人都可以指导我。谢谢。

Answer 1

在训练期间，您可以使用精度层的“top_k”参数计算“top k”精度。这是一个修改bvlc_reference_caffetrain_val.prototxt中现有准确度层的示例：

layer {
  name: "accuracy_top_5"
  type: "Accuracy"
  bottom: "fc8"
  bottom: "label"
  top: "accuracy_top_5"
  accuracy_param {
    top_k: 5
  }
  include {
    phase: TEST
  }
}