Yolov9调用COCOAPI生成APs,APm,APl

最近在做小目标检测的东西,因为后期毕业论文需要,所以开始使用Yolov9模型,运行val.py的时候不会自己产生小目标的AP指标,所以研究了一下,步骤非常简单:
第一步:
在数据集中生成json格式的Annotation文件,存放在数据集目录下:

②找到val.py文件中的save-json将其设置default = True

parser.add_argument('--save-json',default=True, action='store_true', help='save a COCO-JSON results file')

③将下面一行代码进行注释

opt.save_json |= opt.data.endswith('coco.yaml')

④将这一行代码进行替换

# anno_json = str(Path(data.get('path', '../coco')) / 'annotations/instances_val2017.json')  
anno_json = str(Path(data.get('path', '../coco')) / 'annotations.json')  # annotations json

然后运行val.py即可调用COCOeval在验证时添加小目标的指标

以下指标很低的原因只是因为我为了测试能成功训练的时候只跑了一个epoch,这个方法是绝对没有问题的。
补一张AP指标高一点的图:


--------------------------------------------------------------------------------------------------------------------------------
如何生成json文件参考我的这一篇博文:小目标的检测指标APs怎么获得?------数据集Yolo格式生成json文件_yolo转json-CSDN博客

  • 9
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值