YOLOV5 COCO评估指标small、medium、large 自定义概念

您所在的位置:网站首页 coco的总部在哪里 YOLOV5 COCO评估指标small、medium、large 自定义概念

YOLOV5 COCO评估指标small、medium、large 自定义概念

2024-07-16 17:10| 来源: 网络整理| 查看: 265

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 在这里插入图片描述 最后运行结果: 在这里插入图片描述



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3