annotations（COCO 2017）get-id-name-yamlid.py 物体类别 name id对应关系

import json # COCO数据集的路径和注释文件名称 dataDir = 'D:/1. huixiangluo/1_dataset' annFile = '{}/annotations/instances_train2017.json'.format(dataDir) cocoYamlFile = 'coco.yaml' # 读取注释文件 with open(annFile, 'r') as f: ann_data = json.load(f) # 获取所有物体类别的ID号和名称 categories = {} for category in ann_data['categories']: categories[category['id']] = category['name'] # 打印所有物体类别的ID号和名称 # for id, name in categories.items(): # print("ID: {}, Name: {}".format(id, name)) import yaml # 加载coco.yaml文件，获取ID号和对应名称的对应关系 with open(cocoYamlFile, 'r',encoding='utf-8') as f: coco_yaml = yaml.safe_load(f) # 获取ID号和对应名称的对应关系 name_mapping = coco_yaml['names'] id_mapping = {v : k for k, v in name_mapping.items()} # 打印所有物体类别的ID号、名称和在coco.yaml中的对应关系 for id, name in categories.items(): coco_name = id_mapping[name] print("ID: {}, Name: {}, Coco Name: {}".format(id, name, coco_name))