【解决方法】labelme将json文件转png文件

水深无声 2022-11-08 11:58 278阅读 0赞

# 第一步：在你的IDE中创建一个.py文件 #

> 将py文件中的路径修改成你的json文件的路径

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():
        json_file='F:/Anaconda/data_json/'	#这里填入你的json文件的文件夹路径
        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()

# 第二步：运行该py文件 #

*  运行文件如果没问题，那么就会自动将每一个json文件变成一个文件夹，其中包含五个文件，表明转变成功，如图：  
    ![在这里插入图片描述][2021031316495037.png]
 *  如果出现以下报错：

AttributeError: module 'labelme.utils' has no attribute 'draw_label'

解决放方法：使用`Anaconda Prompt`运行以下代码来安装3.16.7版本的labelme

pip install labelme==3.16.7

安装好后再运行py文件应该就没问题啦

[2021031316495037.png]: /images/20221023/75adeb988f2b4323bbe144bfcb9b2bb6.png