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)

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐