调用coco的API来生成coco格式的检测结果

anno_json = '/data/guchengsong/detection/thyroid/raw_data/annotations/test.json'
pred_json = '/home/gcs/thyroid/yolov8-main_g/YOLOv8n_output_val8_json_file.json'
eval_classes = [0]
#try:  # https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocoEvalDemo.ipynb
from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval
import numpy as np
# dataloader = create_dataloader(data[task], imgsz, batch_size, gs, opt, pad=0.5, rect=True,
#                                    prefix=colorstr(f'{task}: '))[0]
anno = COCO(anno_json)  # init annotations api
pred = anno.loadRes(pred_json)  # init predictions api
eval = COCOeval(anno, pred, 'bbox')
# 设置仅评估指定类别ID
eval.params.catIds = eval_classes
# if is_coco:
#     eval.params.imgIds = [int(Path(x).stem) for x in dataloader.dataset.img_files]  # image IDs to evaluate
eval.evaluate()
eval.accumulate()
eval.summarize()
map50, map95 = eval.stats[:2]  # update results (mAP@0.5:0.95, mAP@0.5)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值