python实现批量json文件转txt文件

1.代码

        1）看注释改文件路径

        2）改标签类别和名称

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'] == 'bitter':   #标签名
            txt.write('0')  #标签类别
            txt_write(x, img_x, img_y, txt)

        if x['label'] == 'stem':   #同上
            txt.write('1')
            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 = 'D:/python/projects/yolov5-7.0/datasets/newbitter/labels/val_json'
save_dir = 'D:/python/projects/yolov5-7.0/datasets/newbitter/labels/val'
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

2.转换效果

        1）json文件

        2）转换后txt文件 

 

 3.发布内容是想记录自己的学习过程和经历，更是为了方便自己日后使用。

    如若侵权，联系必删！！！

    如若侵权，联系必删！！！

    如若侵权，联系必删！！！