Yolov3 mAp计算，吐血整理，含Bug处理

Yolov3 mAp计算，吐血整理，含Bug处理

用了两天时间把计算mAp值该踩的坑基本上踩了一遍，整理一下，以免以后忘记

一.所需材料

训练好的权重weights
验证集图片
验证集的xml格式标记
Fastrcnn的voc_eval.py下载地址：
https://github.com/rbgirshick/py-faster-rcnn/blob/master/lib/datasets/voc_eval.py
python2环境

二.步骤

(1)首先运行以下命令

 ./darknet detector valid cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_900.weights -out " " -thresh .5 

把其中的weights改为你自己的weights名

在result中生成了几个以你的class为名的txt文件

里面大致长这样

如果发现文本里是空的，那么检查你的yolov3文件，把里面的train注销掉，改成test

(2)新建python文件 compute_mAp.py 然后在里面输入以下代码

from voc_eval import voc_eval

rec,prec,ap = voc_eval('/home/casiie/yolo/harbor.txt', '/home/casiie/yolo/vallab/{}.xml', '/home/casiie/yolo/val.txt', 'harbor', '.')

print('rec',rec)
print('prec',prec)
print('ap',ap)

这里详细解释下各个路径：
/home/casiie/yolo/harbor.txt 刚才生成的txt文件
/home/casiie/yolo/vallab/{}.xml 存储的验证集xml文件夹
/home/casiie/yolo/val.txt 验证图片的名字txt 这个长这样，注意里面只有文件名，没有路径

'harbor’这个是你要检测的类名
‘.’ 留着，eval_voc.py至少输入四个值

（3）运行，如果不出意外的话就能显示出ap值了

注意：每次计算完ap值要记得把annots.pkl删除再重新编译

mAP（airplane，ship，car）= [AP（airplane）+AP（ship）+AP（car）] / 3

三.错误解决

(1)

Traceback (most recent call last):
  File "computer_mAp.py", line 1, in 
    from voc_eval import voc_eval
  File "/home/casiie/文档/yolot2/voc_eval.py", line 111
    print 'Reading annotation for {:d}/{:d}'.format(
                                           ^
SyntaxError: invalid syntax

原因：没有使用python2环境
解决办法：安装python2.7 推荐使用虚拟环境

(2)
检查eval_voc的

#obj_struct['pose'] = obj.find('pose').text

因为有的xml文件没有pose这一栏，如果不注释掉会报错

参考：

https://blog.csdn.net/zhuoyuezai/article/details/78494126
https://blog.csdn.net/hongxingabc/article/details/80090736
https://blog.csdn.net/LeeWanzhi/article/details/79690275
https://blog.csdn.net/lucky__ing/article/details/78310510#commentBox
https://blog.csdn.net/cgt19910923/article/details/80524173#comments
https://blog.csdn.net/Gentleman_Qin/article/details/84800188#commentsedit
https://blog.csdn.net/yinhuan1649/article/details/82258703#commentBox