Caffe如何得到精确的准确率

我运行完caffe的代码之后只有一个accuracy,我想得到更加精确的概率,比如有3个种类,可以得到A的准确率是30%,B的准确率是30%,C的准确率是40%。这个功能如何实现?Caffe库函数里有这个功能吗?
已邀请:

Machine

赞同来自:

在accuracy_layer中修改代码

雨叶清霜

赞同来自:

楼上说的对。accuracy_layer继承于loss_layer.cpp,在accuracy_layer.cpp中,对其setup函数重写,将top,reshape为3个。然后在计算accuracy的时候,根据lablel不同做出三个分支对应top中三个值就好

要回复问题请先登录注册