labelme标注的图像分割json文件转图片格式

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)