利用COCOAPI计算Yolov3训练出的模型的MAP值,复现ap

 Average Precision  (AP) @[ IoU=0.50:0.95 | area=   all | maxDets=100 ] = 0.314
 Average Precision  (AP) @[ IoU=0.50      | area=   all | maxDets=100 ] = 0.559
 Average Precision  (AP) @[ IoU=0.75      | area=   all | maxDets=100 ] = 0.318
 Average Precision  (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.142
 Average Precision  (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.341
 Average Precision  (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.464
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets=  1 ] = 0.278
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets= 10 ] = 0.420
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets=100 ] = 0.442
 Average Recall     (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.240
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.482
 Average Recall     (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.611

以上是我利用官方给出的yolov3.weights通过计算5k.txt里图片计算出的ap值,已经很贴近论文里作者给出的分辨率为416x416的前提下的31%的值。计算步骤如下:

1、确定测试集:https://pjreddie.com/media/files/coco/5k.part内部含有val2014的5000张图片

2、通过darknet自带的valid命令计算生成被比较的json文件

3、./darknet detector valid cfg/coco.data cfg/yolov3.cfg backup/yolov3.weight注意运行此命令的时候coco.data里的必须添加eval = coco不然生成的文件是class类个txt文件,结果会自动生成在result文件夹下。

4、利用以下python脚本计算ap值

#-*- coding:utf-8 -*-
import matplotlib.pyplot as plt 
from pycocotools.coco import COCO 
from pycocotools.cocoeval import COCOeval 
import numpy as np 
import skimage.io as io 
import pylab,json 
pylab.rcParams['figure.figsize'] = (10.0, 8.0) 
def get_img_id(file_name): 
    ls = [] 
    myset = [] 
    annos = json.load(open(file_name, 'r')) 
    for anno in annos: 
      ls.append(anno['image_id']) 
    myset = {}.fromkeys(ls).keys() 
    return myset 
if __name__ == '__main__': 
    annType = ['segm', 'bbox', 'keypoints']#set iouType to 'segm', 'bbox' or 'keypoints'
    annType = annType[1] # specify type here
    cocoGt_file = '/home/ss/data/coco2014/annotations/instances_val2014.json'
    cocoGt = COCO(cocoGt_file)#取得标注集中coco json对象
    cocoDt_file = '/home/ss/darknet/results/coco_results.json'
    imgIds = get_img_id(cocoDt_file) 
    print len(imgIds)
    cocoDt = cocoGt.loadRes(cocoDt_file)#取得结果集中image json对象
    imgIds = sorted(imgIds)#按顺序排列coco标注集image_id
    imgIds = imgIds[0:5000]#标注集中的image数据
    cocoEval = COCOeval(cocoGt, cocoDt, annType) 
    cocoEval.params.imgIds = imgIds#参数设置
    cocoEval.evaluate()#评价
    cocoEval.accumulate()#积累
    cocoEval.summarize()#总结

改代码添加的路径信息需要根据自己的实际路径添加,计算出来的信息如顶部。

输出信息的具体含义可以参考链接:

https://blog.csdn.net/u014734886/article/details/78831884

https://blog.csdn.net/u014734886/article/details/78831382

  • 4
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 26
    评论
YOLOv5 训练过程中,可以使用以下指令来得到训练结果中的 AP mAP : ``` python train.py --data coco.yaml --cfg models/yolov5s.yaml --weights '' --batch-size 64 --epochs 300 --device 0,1,2,3 --multi-scale --cache-images --name yolov5s_results --project yolov5_results --exist-ok --evolve ``` 其中,`--data` 指定数据集的配置文件,`--cfg` 指定模型的配置文件,`--weights` 指定预训练模型的权重文件,`--batch-size` 指定每个批次的图像数量,`--epochs` 指定训练的轮数,`--device` 指定训练使用的 GPU 设备编号,`--multi-scale` 表示使用多尺度训练,`--cache-images` 表示将图像缓存到内存中以加快训练速度,`--name` 指定训练结果保存的文件夹名称,`--project` 指定训练结果保存的项目名称,`--exist-ok` 表示如果训练结果文件夹已存在,则不会报错,而是继续训练,`--evolve` 表示进行超参数优化。 在训练完成后,可以使用以下指令来计算 AP mAP : ``` python val.py --data coco.yaml --weights runs/train/yolov5s_results/weights/best.pt --task val --img-size 640 --iou 0.65 --half --conf 0.001 --project yolov5_results --name yolov5s_results --exist-ok ``` 其中,`--data` 指定数据集的配置文件,`--weights` 指定训练完成后保存的模型权重文件,`--task` 指定需要进行的任务,`--img-size` 指定输入图像的尺寸,`--iou` 指定 IoU,`--half` 表示使用半精度浮点数计算,`--conf` 指定预测框的置信度阈,`--project` 指定训练结果保存的项目名称,`--name` 指定训练结果保存的文件夹名称,`--exist-ok` 表示如果结果文件夹已存在,则不会报错,而是覆盖其中的文件。 计算完成后,AP mAP 将会在控制台输

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 26
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值