跑模型——labelme的json文件转成yolo使用的txt文件（语义分割）

news/2024/7/10 23:59:41 标签: json, YOLO

前言
将labelme多边形标注的json文件转换成yolo使用的txt文件

import os
import json
import numpy as np
from tqdm import tqdm

#实现函数
def json2txt(path_json, path_txt):  # 可修改生成格式
    with open(path_json, 'r') as path_json:
        jsonx = json.load(path_json)
        with open(path_txt, 'w+') as ftxt:
            width,height = jsonx['imageWidth'],jsonx['imageHeight']
            for shape in jsonx['shapes']:
                label_name = shape['label']
                label_index = str(classes_name.index(label_name))
                label = label_index + ' '
                xy = np.array(shape['points'])
                strxy = ''
                for x, y in xy:
                    x = float(x)/width
                    y = float(y)/height
                    strxy += str(x) + ' ' + str(y) + ' '

                label += strxy
                ftxt.writelines(label + "\n")

dir_json = 'D:/data_val/phone_1_dai/new/output/'  # json存储的文件目录 这里需要在最后加'/'
dir_txt = 'D:/data_val/phone_1_dai/new/txt'  # txt存储目录

#如果不存在则创建txt存储目录
os.makedirs(dir_txt,exist_ok=True)
list_json = os.listdir(dir_json)

#标签名
classes_name = ["_background_", 'phone']

pbar = tqdm(total=len(list_json))
error_list=[]
for cnt, json_name in enumerate(list_json):
    try:
        # print('cnt=%d,name=%s' % (cnt, json_name))
        path_json = dir_json + json_name
        path_txt = dir_txt +'/'+ json_name.replace('.json', '.txt')
        json2txt(path_json, path_txt)
    except Exception as e:
        # print('cnt=%d,name=%s' % (cnt, json_name))
        error_list.append({json_name:e})
    pbar.update(1)


for error in error_list:
    print(error)
print(f"error length={len(error_list)}")