global ANN_ID

coco_annotations = []

for each_ann in labelme['shapes']: # 遍历该json文件中的所有标注

    if each_ann['shape_type'] == 'rectangle': # 筛选出框

        ## 该框的元数据
        bbox_dict = {}
        # 该框的类别 信息
        bbox_dict['category_id'] = label2id[each_ann['label']]
        bbox_dict['segmentation'] = []
        
        bbox_dict['iscrowd'] = 0
        bbox_dict['image_id'] = image_id
        bbox_dict['id'] = ANN_ID
        ANN_ID += 1

        # 获取框坐标
        bbox_left_top_x = min(int(each_ann['points'][0][0]), int(each_ann['points'][1][0]))
        bbox_left_top_y = min(int(each_ann['points'][0][1]), int(each_ann['points'][1][1]))
        bbox_right_bottom_x = max(int(each_ann['points'][0][0]), int(each_ann['points'][1][0]))
        bbox_right_bottom_y = max(int(each_ann['points'][0][1]), int(each_ann['points'][1][1]))
        bbox_w = bbox_right_bottom_x - bbox_left_top_x
        bbox_h = bbox_right_bottom_y - bbox_left_top_y
        bbox_dict['bbox'] = [bbox_left_top_x, bbox_left_top_y, bbox_w, bbox_h] # 左上角x、y、框的w、h
        bbox_dict['area'] = bbox_w * bbox_h
        
        # 筛选出分割多段线
        for each_ann in labelme['shapes']: # 遍历所有标注
            if each_ann['shape_type'] == 'polygon': # 筛选出分割多段线标注
                # 第一个点的坐标
                first_x = each_ann['points'][0][0]
                first_y = each_ann['points'][0][1]
                if (first_x>bbox_left_top_x) & (first_x<bbox_right_bottom_x) & (first_y<bbox_right_bottom_y) & (first_y>bbox_left_top_y): # 筛选出在该个体框中的关键点
                    bbox_dict['segmentation'] = list(map(lambda x: list(map(lambda y: round(y, 2), x)), each_ann['points'])) # 坐标保留两位小数
                
        coco_annotations.append(bbox_dict)
        
return coco_annotations`