Labelme---批量json转png

原创

已于 2024-07-04 12:24:31 修改 · 366 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#json

于 2024-07-04 12:23:27 首次发布

我们原始的labelme的有转png的脚本的。

比如

labelme_json_to_dataset H:\\dataset\\train\\label\\1.json

但是只能转一个，所以需要修改这个脚本，找到这个脚本，我的位置在D:\Anaconda\Lib\site-packages\labelme\cli\json_to_dataset.py，如果，找不到，可以利用everything这个软件查找。

然后修改代码如下

import argparse
import base64
import json
import os
import os.path as osp
import imgviz
import PIL.Image
import yaml
from labelme.logger import logger
from labelme import utils

'''multiple json files'''
def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('json_file')   # 标注文件json所在的文件夹
    parser.add_argument('-o', '--out', default=None)
    args = parser.parse_args()

    json_file = args.json_file

    list = os.listdir(json_file)   # 获取json文件列表

    for i in range(0, len(list)):
        path = os.path.join(json_file, list[i])  # 获取每个json文件的绝对路径
        filename = list[i][:-5]       # 提取出.json前的字符作为文件名，以便后续保存Label图片的时候使用
        extension = list[i][-4:]
        if extension == 'json