批量将.json文件转换成img.png;label.png;label_name.txt;label_viz.pgn;info.yaml文件

  1. 调出终端(cmd命令)。

  2. 首先激活安装有labelme的环境。(我的在python35环境里)

  3. 在终端,打开要存储img.png;label.png;label_name.txt;label_viz.pgn;info.yaml文件的文件夹。(我的存放在G:\Anconda3\data_json文件夹下)
    这里写图片描述

  4. 把(G:\anconda3\envs\python35\lib\site-packages\labelme\cli\json_to_dataset.py)文件程序替换成以下内容:

  5. 替换完成后,在终端输入labelme_json_to_dataset + (.json文件的文件夹)
    这里写图片描述

  6. 运行结果:
    批量将.json文件转换成img.png;label.png;label_name.txt;label_viz.pgn;info.yaml文件_第1张图片

  7. **备注:
    代码来自https://blog.csdn.net/weixin_41831559/article/details/80835913

python程序

import argparse
import json
import os
import os.path as osp
import warnings
import numpy as np
import PIL.Image
import yaml
from labelme import utils
def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('json_file')
    parser.add_argument('-o', '--out', default=None)
    args = parser.parse_args()
    json_file = args.json_file
    list = os.listdir(json_file) 
    for i in range(0, len(list)):
        path = os.path.join(json_file, list[i])
        if os.path.isfile(path):
            data = json.load(open(path))
            img = utils.img_b64_to_arr(data['imageData'])
            lbl, lbl_names = utils.labelme_shapes_to_label(img.shape, data['shapes'])
            captions = ['%d: %s' % (l, name) for l, name in enumerate(lbl_names)]
            lbl_viz = utils.draw_label(lbl, img, captions)
            out_dir = osp.basename(list[i]).replace('.', '_')
            out_dir = osp.join(osp.dirname(list[i]), out_dir)
            if not osp.exists(out_dir):
                os.mkdir(out_dir)
            PIL.Image.fromarray(img).save(osp.join(out_dir, 'img.png'))
            PIL.Image.fromarray(lbl).save(osp.join(out_dir, 'label.png'))
            PIL.Image.fromarray(lbl_viz).save(osp.join(out_dir, 'label_viz.png'))
            with open(osp.join(out_dir, 'label_names.txt'), 'w') as f:
                for lbl_name in lbl_names:
                    f.write(lbl_name + '\n')
            warnings.warn('info.yaml is being replaced by label_names.txt')
            info = dict(label_names=lbl_names)
            with open(osp.join(out_dir, 'info.yaml'), 'w') as f:
                yaml.safe_dump(info, f, default_flow_style=False)
            print('Saved to: %s' % out_dir)
if __name__ == '__main__':
    main()

你可能感兴趣的:(python)