YOLOv5 json2txt

import os
import cv2
import json
import numpy as np


def txt_write(x,img_x,img_y,txt):
    data = x['points']
    n = 1
    for x in data:
        for i in x:
            if n % 2 == 0:
                txt.write(' ' + str(round(i/img_x,6)))
                n += 1
            else:
                txt.write(' ' + str(round(i/img_y,6)))
                n += 1  
    txt.write('\n') 


def json2txt(json_path,save_path):
    txt = open(save_path,'w')
    with open(json_path, "r") as f:
        data = f.read()
    data = json.loads(data)
    img_x = data['imageHeight']
    img_y = data['imageWidth']
    shapes = data['shapes']

    for x in shapes:
        #print(x['label'])
        #此处面向不同分类，需要改动下面的标签值，如果是多分类，那么需要增加新的if
        #只是单分类的话，可以直接去掉if，把里面的模块拿出来用

        if x['label']=='outlooking':
            txt.write('0') 
            txt_write(x,img_x,img_y,txt)
    txt.close()
    
#单文件测试
# save_dir = "/workspace/" #文件路径
# name = 'test'
# save_path = save_dir + name + '.txt' # 也可以是.doc
# json_path = '/json/65161.json'
# json2txt(json_path,save_path)

#文件夹
json_dir = '/workspace/data/json'
save_dir = 'workspace/data/txt'
files = os.listdir(json_dir)
os.makedirs(save_dir, exist_ok=True)
num = 1
for file in files :
    name = file[0:-5]
    json_path = json_dir + '/' + name + '.json'
    save_path = save_dir + '/' + name + '.txt'
    json2txt(json_path,save_path)
    print(num,'/',len(files),':',name)
    num += 1