labelme将json文件批量转换 json_to_dataset的方法

最新推荐文章于 2022-12-14 11:08:30 发布

奋斗的猪猪霞

最新推荐文章于 2022-12-14 11:08:30 发布

阅读量1.1k

点赞数 1

文章标签： python

本文链接：https://blog.csdn.net/weixin_47286519/article/details/116993802

版权

本文介绍了如何批量将json文件通过labelme转换成dataset。主要步骤包括：修改json_to_dataset.py文件，找到json_to_dataset.exe路径，激活labelme环境并在cmd中执行转换命令，最后处理可能出现的错误，如版本不兼容问题。

摘要由CSDN通过智能技术生成

1、修改 json_to_dataset.py

先找到 json_to_dataset.py的地址，我的路径是

D:\anaconda\envs\labelme\Lib\site-packages\labelme\cli\json_to_dataset.py

然后打开json_to_dataset.py，将代码更改为

import argparse
import json
import os
import os.path as osp
import warnings

import PIL.Image
import yaml

from labelme import utils
import base64


def main():
    warnings.warn("This script is aimed to demonstrate how to convert the\n"
                  "JSON file to a single image dataset, and not to handle\n"
                  "multiple JSON files to generate a real-use dataset.")
    parser = argparse.ArgumentParser()
    parser.add_argument('json_file')
    parser.add_argument('-o', '--out', default=None)
    args = parser.parse_args()

    json_file = args.json_file
    if args.out is None:
        out_dir = osp.basename(json_file).replace('.', '_')
        out_dir = osp.join(osp.dirname(json_file), out_dir)
    else:
        out_dir = args.out
    if not osp.exists(out_dir)