label标注的标签是json格式,会对选择的区域进行打点标记,所有的点围成不规则的多边形区域,中间的部分就是分割的区域
这里编写了python脚本,可以把json文件转为图片格式,这里是png格式
目标摆放如下
转化结果如下:
代码如下:
import os
import cv2
import numpy as np
from tqdm import tqdm
import json
root = 'label'
leibie = [os.path.join(root,i) for i in os.listdir(root)]
for i in tqdm(leibie):
data = open(i,encoding='utf8')
data = json.load(data)
a = np.array(data['shapes'][0]['points']).astype(np.int32)
h = data['imageHeight']
w = data['imageWidth']
ret_img = np.zeros((h,w,3),dtype=np.int32)
ret_path = i.replace('.json','.png').replace('labels','png')
ret_img = cv2.fillPoly(ret_img,[a],color=(255,255,255))
ret_img = ret_img.astype(np.uint8)
ret_img =cv2.cvtColor(ret_img,cv2.COLOR_BGR2GRAY)
cv2.imwrite(ret_path,ret_img)