DBnet，json标签转换txt

import json
import os

# 指定包含json文件的目录
input_directory = 'E:/Text_traffic_signs/labelme'

# 指定转换后的txt文件要保存在哪个目录
output_directory = 'E:/Text_traffic_signs/labelme-gt'

# 确保输出目录存在
if not os.path.exists(output_directory):
    os.makedirs(output_directory)

# 遍历目录中的文件
for filename in os.listdir(input_directory):

    # 检查文件是否是json文件
    if filename.endswith('.json'):

        # 打开并读取json文件
        with open(os.path.join(input_directory, filename)) as f:
            data = json.load(f)

        base = os.path.splitext(filename)[0]
        # 打开txt文件并写入
        with open(os.path.join(output_directory, base + '.txt'), 'w') as f:
            for shape in data['shapes']:
                label = shape['label']
                points = shape['points']
                # 将points中的每个元素转化为字符串，并用逗号连接
                points_str = ','.join([str(point) for point in points])
                text = points_str.replace("[", "").replace("]", "")
                # 将label和points写入到txt文件中，每个用换行符隔开
                # f.write(label + '\n')
                f.write(text + "," + label + '\n')