import os, cv2, json
import numpy as np
classes = ['road']
base_path = 'E:\\aaa\\dataf1\\'
path_list = [i.split('.')[0] for i in os.listdir(base_path)]
for path in path_list:
image = cv2.imread(f'{base_path}\\{path}.jpg')
h, w, c = image.shape
with open(f'{base_path}\\{path}.json') as f:
masks = json.load(f)['shapes']
with open(f'{base_path}\\{path}.txt', 'w+') as f:
for idx, mask_data in enumerate(masks):
mask_label = mask_data['label']
if '_' in mask_label:
mask_label = mask_label.split('_')[0]
mask = np.array([np.array(i) for i in mask_data['points']], dtype=np.float)
mask[:, 0] /= w
mask[:, 1] /= h
mask = mask.reshape((-1))
if idx != 0:
f.write('\n')
f.write(f'{classes.index(mask_label)} {" ".join(list(map(lambda x:f"{x:.6f}", mask)))}')
标签：yolov5,mask,label,json,base,path,np,格式 From： https://www.cnblogs.com/fzx6/p/16988220.html