c++ - Caffe で上位 5 つのエラー率を計算していますか?

Question

kソフトマックス出力からソートトップ予測を計算するために synset を使用します。

これにより、上位5つのクラス名が得られます。しかし、その割合を計算する方法を知りたいです。つまり、上位 5% の誤差です。

誰でも親切に私を案内してもらえますか。ありがとう。

Answer 1

トレーニング中に、「top_k」パラメーターを精度レイヤーに使用して「上位 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
  }
}