【mmdetection】验证时AP为-1

小涛44

已于 2023-04-21 10:08:08 修改

阅读量3.1k

点赞数 13

分类专栏： Bug和黑科技杂记文章标签：人工智能目标检测 python

于 2023-03-20 15:54:43 首次发布

本文链接：https://blog.csdn.net/qq_56077562/article/details/129669046

版权

Bug和黑科技杂记专栏收录该内容

4 篇文章

订阅专栏

问题

AP的解释

可能的Bug

问题

验证的时候评价指标输出是酱紫的

Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.177
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=1000 ] = 0.608
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=1000 ] = 0.057
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=1000 ] = 0.293
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=1000 ] = -1.000
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=1000 ] = -1.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.292
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=300 ] = 0.292
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=1000 ] = 0.292
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=1000 ] = 0.292
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=1000 ] = -1.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=1000 ] = -1.000

可见area= medium和area=large的AP值为-1，不知道是不是bug导致我的mAP很低，翻了一圈找到了以下解释，总算是弄懂了

AP的解释

首先得搞懂这些AP都代表什么意思，为什么就medium和large的AP值为-1

所以说，这和验证集里标注框面积有关，得检查自己的数据集如果数据框的都是比较小的，中型和大型的自然就没有，所以无法评判（从Github上的回答来看，源码里对于没有的就设置为-1）

可能的Bug

那如果说确实有中大型的检测框，还是-1，那是为啥呢？

那可能是coco数据集的json文件有问题了，在json2coco.py转换时area恒等于1了

因此找到并修改

# annotation['area'] = 1.0  # 导致medium和large的ap都是-1
annotation['area'] = annotation['bbox'][2] * annotation['bbox'][3]

就可以了~