MMdetection2测试voc数据获取precision

文章目录[隐藏]

在使用mmdetection时,测试voc数据集无法得到精度指标,只有recall和mAP,这里可以通过修改几行代码来获得precision指标

修改后打印结果:

代码修改

修改mmdet/core/evaluation/mean_ap.py文件里的print_map_summary函数的5处代码,你可以直接复制这个函数,然后替换掉原来的函数。修改后的函数为:

def print_map_summary(mean_ap,
                      results,
                      dataset=None,
                      scale_ranges=None,
                      logger=None):
    """Print mAP and results of each class.

    A table will be printed to show the gts/dets/recall/AP of each class and
    the mAP.

    Args:
        mean_ap (float): Calculated from `eval_map()`.
        results (list[dict]): Calculated from `eval_map()`.
       

版权声明:本文为CSDN博主「TC1398」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/m0_61102347/article/details/122310693

TC1398

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

YOLOX笔记

目录 1. 样本匹配 正负样本划分过程 2. yoloxwarmcos 学习率 3. 无法开启多gpu训练, 或者多gpu训练卡住? 1. 样本匹配 正负样本划分过程 说明: gt_centerbbox是在gt_bbox中心点向四周