[人工智能] json文件格式转换为png文件格式

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 人工智能 -> json文件格式转换为png文件格式 -> 正文阅读

[人工智能]json文件格式转换为png文件格式

话不多少，直接上代码，更换源目录和目标目录即可

1、导入库

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

json_dir = 'G:/json_filedir/'    #json文件所在文件夹(注：文件夹中只能包含json文件)
label_dir = 'G:/label_filedir/'   #目标输出文件夹

3、定义转换函数

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
        cv2.imwrite(png_save_path, label_png)#将label.png文件重新保存在目标目录下
        shutil.rmtree(temporary_path)#每次循环执行完后删除生成的临时文件，不删的话，再次执行时需要手动删除

4、执行函数，转换格式

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

5、查看效果

test_img = cv2.imread('G:/test.jpg')
test_label = cv2.imread('G:/label_filedir/test.png')

plt.figure(figsize=(16,8))
plt.subplot(121),plt.imshow(test_img,'gray')
plt.subplot(122),plt.imshow(test_label,'gray')
plt.show()