计算 Caffe 中的前 5 错误率？

Question

我使用 synset 从 softmax 输出计算排序前 k 预测。

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

谁能指导一下。 谢谢。

Answer 1

在训练期间，您可以使用 "top_k" 参数到精度层来计算 "top k" 精度。这是修改 bvlc_reference_caffe's 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
  }
}