coco数据集json

你可以使用以下步骤将COCO数据集的JSON格式转换为YOLO格式：

1. 导入所需的库：

import json

2. 读取COCO JSON文件：

with open('path/to/coco.json', 'r') as f:
    data = json.load(f)

3. 定义类别标签的映射关系：

class_labels = {
    1: 'person',
    2: 'bicycle',
    3: 'car',
    # 添加其他类别...
}

4. 遍历COCO数据集中的每个图像和标注：

for image in data['images']:
    image_id = image['id']
    file_name = image['file_name']
    
    # 根据图像ID获取对应的标注信息
    annotations = [ann for ann in data['annotations'] if ann['image_id'] == image_id]
    
    # 创建YOLO格式的标注文件
    with open('path/to/yolo_annotations/{}.txt'.format(file_name.split('.')[0]), 'w') as f:
        for ann in annotations:
            category_id = ann['category_id']
            bbox = ann['bbox']
            
            # 计算YOLO格式的边界框坐标
            x_center = bbox[0] + bbox[2] / 2
            y_center = bbox[1] + bbox[3] / 2
            width = bbox[2]
            height = bbox[3]
            
            # 将边界框坐标归一化到图像尺寸范围（0~1）
            x_center /= image['width']
            y_center /= image['height']
            width /= image['width']
            height /= image['height']
            
            # 将标注写入YOLO格式的标注文件
            f.write('{} {:.6f} {:.6f} {:.6f} {:.6f}\n'.format(class_labels[category_id],