如何批量格式化机械Turk边界框数据？

import csv
import json

# 假设CSV文件名为input.csv，格式如下：
# image_id,xmin,ymin,xmax,ymax
# 1,100,100,200,200
# 2,150,150,250,250

# 读取CSV文件并转换为JSON格式
def convert_csv_to_json(csv_file, json_file):
    data = []
    with open(csv_file, mode='r') as csvfile:
        reader = csv.DictReader(csvfile)
        for row in reader:
            bbox = {
                'image_id': row['image_id'],
                'bbox': [int(row['xmin']), int(row['ymin']), int(row['xmax']), int(row['ymax'])]
            }
            data.append(bbox)
    
    with open(json_file, mode='w') as jsonfile:
        json.dump(data, jsonfile, indent=4)

# 调用函数进行转换
convert_csv_to_json('input.csv', 'output.json')