Faster RCNN源码函数作用学习——voc_eval.py

voc_eval.py :

输出某类别测试结果的rec , prec , ap

一、unique_boxes()函数

def parse_rec(filename):
    ......
  return objects

函数作用：

解析一个PASCAL VOC xml文件。返回值是key为('name', 'pose', 'truncated', 'difficult', 'bbox(xmin, ymin, xmax, ymax)')构成的字典objects

二、voc_ap()函数

def voc_ap(rec, prec, use_07_metric=False):
    .....
  return ap

函数作用：

根据传入的rec, prec, 求ap

三、voc_eval()函数

# Top level function that does the PASCAL VOC evaluation.
def voc_eval(detpath,
             annopath,
             imagesetfile,
             classname,
             cachedir,
             ovthresh=0.5,
             use_07_metric=False):
    ......
  return rec, prec, ap

函数作用：

根据测试结果路径、注释路径（gt）、类别名称等，输出特定类别的rec， prec, ap