COCO评估指标small、medium、large官方默认:
small : 小目标 (框的面积小于32平方像素
medium: 中目标 (32-96 )
large : 大目标 (大于96 )
对用我的数据集来说,进行coco评价时出现:
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1
说明没有小于32的,但是我又想结果有small,所以进行了修改定义。
在val.py文件中找到from pycocotools.cocoeval import COCOeval
按照Ctrl+鼠标点击COCOeval 跳转到 cocoeval.py文件。
在 cocoeval.py文件中, def setDetParams(self)和 def setKpParams(self)中areaRng 对应部分,这里我定义小目标小于64*64
最后运行结果: