如何在python中将CSV表转换为COCO格式？

import csv
import json

csv_file = open('input.csv', 'r')
csv_reader = csv.reader(csv_file)
header = next(csv_reader)  # 获取CSV文件的表头

data = []
for row in csv_reader:
    data.append(row)

coco_data = {
    "info": {},
    "licenses": [],
    "categories": [],
    "images": [],
    "annotations": []
}

image_id = 1
annotation_id = 1

for row in data:
    image = {
        "id": image_id,
        "file_name": row[0],  # 图像文件名
        "width": int(row[1]),  # 图像宽度
        "height": int(row[2])  # 图像高度
    }
    coco_data["images"].append(image)

    annotation = {
        "id": annotation_id,
        "image_id": image_id,
        "category_id": int(row[3]),  # 目标类别ID
        "bbox": [int(row[4]), int(row[5]), int(row[6]), int(row[7])]  # 边界框坐标
    }
    coco_data["annotations"].append(annotation)

    image_id += 1
    annotation_id += 1

with open('output.json', 'w') as json_file:
    json.dump(coco_data, json_file)