Json转二值图像

Json文件通过labelme进行标识

image路径
G:\PyCharm\workspace\unet_42-master\datasets\label_bz\test\image

label路径
G:\PyCharm\workspace\unet_42-master\datasets\label_bz\test\label

待转换路径
G:\PyCharm\workspace\unet_42-master\datasets\label_bz\test\mask

代码如下：

# -*- coding: UTF-8 -*-

from __future__ import print_function
import argparse
import glob
import math
import json
import os
import os.path as osp
import shutil
import numpy as np
import PIL.Image
import PIL.ImageDraw
import cv2


def json2png(json_folder, png_save_folder):
    if osp.isdir(png_save_folder):
        shutil.rmtree(png_save_folder)
    os.makedirs(png_save_folder)
    json_files = os.listdir(json_folder)
    for json_file in json_files:
        json_path = osp.join(json_folder, json_file)
        os.system("labelme_json_to_dataset {}".format(json_path))
        label_path = osp.join(json_folder, json_file.split(".")[0] + "_json/label.png")
        png_save_path = osp.join(png_save_folder, json_file.split(".")[0] + ".png")
        label_png = cv2.imread(label_path, 0)
        label_png[label_png > 0] = 255
        cv2.imwrite(png_save_path, label_png)
        # shutil.copy(label_path, png_save_path)
        # break


if __name__ == '__main__':
    # !!!!你的json文件夹下只能有json文件不能有其他文件
    json2png(json_folder=r"G:\PyCharm\workspace\unet_42-master\datasets\label_bz\test\label",#json文件所在位置
             png_save_folder=r"G:\PyCharm\workspace\unet_42-master\datasets\label_bz\test\mask")#生成二值图像的位置

运行结果：
这些文件夹中存放打好标签的信息


对应生成的二值图像