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 cv2def 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] = 255cv2.imwrite(png_save_path, label_png)# shutil.copy(label_path, png_save_path)# breakif __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")#生成二值圖像的位置
運行結果:
這些文件夾中存放打好標簽的信息
對應生成的二值圖像