1. 导入库
import cv2
import numpy as np
import os
import shutil
import matplotlib.pyplot as plt
2. 定义方法
def json2png(json_folder, png_save_folder):
if os.path.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 = os.path.join(json_folder, json_file)
os.system("labelme_json_to_dataset {}".format(json_path))
label_path = os.path.join(json_folder, json_file.split(".")[0] + "_json/label.png")
temporary_path = os.path.join(json_folder, json_file.split(".")[0] + "_json")
png_save_path = os.path.join(png_save_folder, json_file.split(".")[0] + ".png")
label_png = cv2.imread(label_path, 0)
cv2.imwrite(png_save_path, label_png)
shutil.rmtree(temporary_path)
3. 调用
json_dir = './BanMaXian/train_json'
label_dir = './BanMaXian/train_label'
json2png(json_folder=json_dir, png_save_folder=label_dir)