计算 Caffe 中的前 5 名错误率？

Question

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

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

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

Answer 1

在训练期间，您可以使用精度层的“top_k”参数计算“top k”精度。这是修改bvlc_reference_caffe 中train_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
  }
}