将json文件转换成png文件,文件名不能有空格,点等

json_dir 是原始的文件夹
label_dir 是转换之后的文件夹

import cv2
import numpy as np
import os
import shutil
import matplotlib.pyplot as plt


json_dir = r'C:\Users\Administrator\Desktop\project\fire_dataset_7_0\0_json'    #json文件所在文件夹(注:文件夹中只能包含json文件)
label_dir = r'C:\Users\Administrator\Desktop\project\fire_dataset_7_0\0_Masks'   #目标输出文件夹

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)#生成单个源json文件的路径
        os.system("labelme_json_to_dataset {}".format(json_path))#生成label文件夹及label.png文件
        label_path = os.path.join(json_folder, json_file.split(".")[0] + "_json/label.png")#生成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)#读取label.png
        print(label_png)

        cv2.imwrite(png_save_path, label_png)#将label.png文件重新保存在目标目录下


        shutil.rmtree(temporary_path)#每次循环执行完后删除生成的临时文件,不删的话,再次执行时需要手动删除

        # 执行函数(批量转换)
json2png(json_folder=json_dir, png_save_folder=label_dir)

Guess you like

Origin blog.csdn.net/qq_44666320/article/details/126678597