json文件转换为png、jpg，标签可视化

import base64
import json
import os
import os.path as osp
from labelme import utils


def main():
 
    json_folder = r"D:\GHY\data\MAY\json_0001"
    out_dir = r"D:\GHY\data\MAY\json_png_mask"
    if not osp.exists(out_dir):
        os.mkdir(out_dir)

    json_list = os.listdir(json_folder)
    for i in range(0, len(json_list)):
        json_file = os.path.join(json_folder, json_list[i])
        out_name = json_list[i][:-5]
        data = json.load(open(json_file))
        imageData = data.get("imageData")

        if not imageData:
            imagePath = os.path.join(os.path.dirname(json_file), data["imagePath"])
            with open(imagePath, "rb") as f:
                imageData = f.read()
                imageData = base64.b64encode(imageData).decode("utf-8")
        img = utils.img_b64_to_arr(imageData)

        label_name_to_value = {"_background_": 0}
        for shape in sorted(data["shapes"], key=lambda x: x["label"]):
            label_name = shape["label"]
            if label_name in label_name_to_value:
                label_value = label_name_to_value[label_name]
            else:
                label_value = len(label_name_to_value)
                label_name_to_value[label_name] = label_value
        lbl, _ = utils.shapes_to_label(
            img.shape, data["shapes"], label_name_to_value
        )

        label_names = [None] * (max(label_name_to_value.values()) + 1)
        for name, value in label_name_to_value.items():
            label_names[value] = name


        #PIL.Image.fromarray(img).save(osp.join(out_dir, "img.png"))
        utils.lblsave(osp.join(out_dir, out_name+".png"), lbl)
        #PIL.Image.fromarray(lbl_viz).save(osp.join(out_dir, "label_viz.png"))



if __name__ == "__main__":
    main()