labelme 图像分割数据标注后格式转换 json-＞txt

#coding=utf-8
import json
import os


def convert_labelme_to_yolo(json_file, output_txt_file, label_to_class_id):
    with open(json_file, 'r') as file:
        data = json.load(file)

    image_width = data['imageWidth']
    image_height = data['imageHeight']

    with open(output_txt_file, 'w') as file:
        for shape in data['shapes']:
            points = shape['points']
            class_id = label_to_class_id.get(shape['label'], -1)
            normalized_points = []
            for x, y in points:
                nx = round(x / image_width, 6)
                ny = round(y / image_height, 6)
                normalized_points.append(f"{nx} {ny}")
            file.write(f"{class_id} " + " ".join(normalized_points) + "\n")


def batch_convert(json_folder, output_folder, label_to_class_id):
    # 确保输出文件夹存在
    os.makedirs(output_folder, exist_ok=True)

    # 遍历文件夹中的所有JSON文件
    for filename in os.listdir(json_folder):
        if filename.endswith