使用caffe自动测试模型top5的结果

方法很简单,直接在定义网络的prototxt里面最后加一层就可以了。
这一层定义如下

layer {
  name: "accuracy_5"
  type: "Accuracy"
  bottom: "fc8_sum"
  bottom: "label"
  top: "accuracy_5"
  accuracy_param
  {
    top_k:5
  }
  include {
    phase: TEST
  }
}
posted @ 2016-02-25 17:54  77695  阅读(2569)  评论(0编辑  收藏  举报