帮我写一个批量将yolopose转换成单个json文件的cvat导出的cocokeypoints的脚本

好的，以下是一个简单的 Python 脚本，可以将 YOLOpose 转换成单个 JSON 文件：

import os
import json

# 定义文件路径
yolo_dir = './yolopose_data'
cvat_dir = './cvat_data'
output_dir = './json_data'

# 遍历 YOLOpose 数据目录
for filename in os.listdir(yolo_dir):
    if filename.endswith('.txt'):
        # 获取文件名（不带扩展名）
        name = os.path.splitext(filename)[0]
        
        # 获取 YOLOpose 数据
        yolo_path = os.path.join(yolo_dir, filename)
        with open(yolo_path, 'r') as f:
            data = f.read()
        
        # 将数据转换成 COCO keypoints 格式
        keypoints = []
        for line in data.splitlines():
            parts = line.split()
            keypoints.append(float(parts[1]))
            keypoints.append(float(parts[2]))
            keypoints.append(2)
        
        # 获取对应的 CVAT 数据
        cvat_path = os.path.join(cvat_dir, name + '.json')
        with open(cvat_path, 'r') as f:
            cvat_data = json.load(f)
        
        # 将 COCO keypoints 添加到 CVAT 数据中
        cvat_data['shapes'][0]['points'] = keypoints
        
        # 写入 JSON 文件
        output_path = os.path.join(output_dir, name + '.json')
        with open(output_path, 'w') as f:
            json.dump(cvat_data, f, indent=2)

希望这个脚本对你有帮助！