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

前言
将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)/widthy = float(y)/heightstrxy += str(x) + ' ' + str(y) + ' 'label += strxyftxt.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_namepath_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)}")