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

时间:2022-05-13 14:56:25

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

这一层定义如下

layer {
name: "accuracy_5"
type: "Accuracy"
bottom: "fc8_sum"
bottom: "label"
top: "accuracy_5"
accuracy_param
{
top_k:5
}
include {
phase: TEST
}
}