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