coco_api调试

最新推荐文章于 2023-11-21 15:23:22 发布

HANgugu

最新推荐文章于 2023-11-21 15:23:22 发布

阅读量195

点赞数

文章标签： python

本文链接：https://blog.csdn.net/hanshixin/article/details/125538169

版权

# coco_api调试
from pycocotools.coco import COCO


val_info = r"D:\COCO\annotations\instances_val2017.json"


coco_api = COCO(val_info)  # 导入验证集
all_ids = coco_api.imgs.keys()
cat_ids = sorted(coco_api.getCatIds())

cat2label = {cat_id: i for i, cat_id in enumerate(cat_ids)}

cats = coco_api.loadCats(cat_ids)

class_names = [cat["name"] for cat in cats]

img_ids = sorted(coco_api.imgs.keys())

img_info = coco_api.loadImgs(img_ids)

print('---------------')