• mAP计算


    import os
    from utils.eval_utils import evaluate_on_cpu, evaluate_on_gpu, get_preds_gpu, voc_eval, parse_gt_rec
    from read_xml import readXMLAndReturnBox,readCheckResultTxt
    
    print('---')
    
    #readXMLAndReturnBox('/home/star/RCNN-20200503/YOLOv3_TensorFlow/data/VOC2007/Annotations/2019_07_14-10_11_34_left.xml')
    
    
    box_1 = readXMLAndReturnBox('/home/star/RCNN-20200503/YOLOv3_TensorFlow/data/VOC2007/Annotations/2019_07_14-09_07_14_left.xml')
    
    #def voc_eval(gt_dict, val_preds, classidx, iou_thres=0.5, use_07_metric=False)
    
    
    #gt_dict['2019_07_14-08_06_34_left.txt'] = box_1
    #gt_dict.setdefault('2019_07_14-08_06_34_left.txt',[]).append(box_1)
    
    #val_preds = [['2019_07_14-09_07_14_left',1199, 412, 1240, 488, 0,0.96,0]]
    
    
    xml_lists = os.listdir('/home/star/RCNN-20200503/YOLOv3_TensorFlow/data/VOC2007/Annotations/')
    txt_lists = os.listdir('/home/star/RCNN-20200503/YOLOv3_TensorFlow/data/detect_result2/')
    
    xml_list = []
    
    '''
    for file_index in xml_lists:
        print(file_index)
    '''
    
    xml_list = [os.path.join('/home/star/RCNN-20200503/YOLOv3_TensorFlow/data/VOC2007/Annotations/',file_index) for file_index in xml_lists]
    txt_list = [os.path.join('/home/star/RCNN-20200503/YOLOv3_TensorFlow/data/detect_result2/',file_index) for file_index in txt_lists]
    
    ap_total =[]
    
    for file_index in range(len(xml_list)):
        gt_dict = {}
        box_1 = readXMLAndReturnBox(xml_list[file_index])
        gt_dict[xml_lists[file_index]] = box_1
    
        val_index  = readCheckResultTxt(txt_list[file_index],xml_lists[file_index])
    
    
        print(gt_dict)
        print(val_index)
    
        npos, nd, rec, prec, ap = voc_eval(gt_dict, val_index, 0, iou_thres=0.5)
        ap_total.append(ap)
    
    t_mAP = 0.00
    
    for index_ap in ap_total:
        t_mAP = t_mAP + index_ap
    
    mAP = t_mAP/len(ap_total)
    
    
    print("="*30)
    print("mAP:")
    print(mAP)
    
    '''
    val_index  = readCheckResultTxt('/home/star/RCNN-20200503/YOLOv3_TensorFlow/data/detect_result2/2019_07_14-08_06_34_left.txt','2019_07_14-08_06_34_left.txt')
    #val_index.insert(0,'2019_07_14-08_08_34_left.txt')
    
    print(val_index)
    print(gt_dict)
    
    
    
    
    #val_preds = [[['2019_07_14-09_07_14_left',929, 428, 973, 479, 0.96,0], ['2019_07_14-09_07_14_left',1199, 412, 1240, 488, 0.96,0], ['2019_07_14-09_07_14_left',1206, 414, 1279, 564, 0.96,0]]]
    #./darknet detector map cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_80172.weights
    
    #gt_dict = {'2019_07_14-08_06_34_left.txt': [[929, 428, 973, 479, 0], [1199, 412, 1240, 488, 0], [1206, 414, 1279, 564, 0]]}
    
    
    
    npos, nd, rec, prec, ap = voc_eval(gt_dict, val_index, 0, iou_thres=0.5)
    print(ap)
    '''
  • 相关阅读:
    发现一个奇怪的问题: 不能把文件取名为 con
    博客园新购服务器硬件配置
    [重要新功能]团队Blog
    [庆祝]博客园已迁至新服务器
    [公告]博客园论坛开放注册
    博客园出现了奇怪的cookie问题
    [公告]博客园聊天室试运行
    [重发]请为你喜欢的博客园杂志的名字投上一票
    [征询意见]准备采用“创作共用”协议保护大家的原创作品
    博客园期刊制作小组Blog开通
  • 原文地址:https://www.cnblogs.com/herd/p/12892232.html
Copyright © 2020-2023  润新知