方法很简单,直接在定义网络的prototxt里面最后加一层就可以了。 这一层定义如下
layer { name: "accuracy_5" type: "Accuracy" bottom: "fc8_sum" bottom: "label" top: "accuracy_5" accuracy_param { top_k:5 } include { phase: TEST } }