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.发布内容是想记录自己的学习过程和经历,更是为了方便自己日后使用。
如若侵权,联系必删!!!
如若侵权,联系必删!!!
如若侵权,联系必删!!!