最近在做语义分割,现需要训练自己标注的数据,使用labelme工具标注图像,但是labelme自带的转换标注json文件一次只能转一个文件,现在改写labelme的源码使得可以一次转多个json文件。在自己编代码前参考了网上的一些博客,是在不得不说,真是辣鸡,各种报错。现附上自己的代码,亲测在python3.7下是可以运行的。改造的代码是在anaconda安装的虚拟环境下lableme安装的文件夹中,如我的是:D:\software\Anaconda\Anconda_env\envs\tf14-py37\Lib\site-packages\labelme,进入这个目录后打开cli文件,里面有个json_to_dataset.py文件,在此文件上改造即可。改造完保存,在终端执行如下指令:
labelme_json _to_dataset json文件夹名 -o 结果保存文件名
import argparse
import base64
import json
import os
import os.path as osp
import imgviz
import PIL.Image
from labelme.logger import logger
from labelme import utils
import yaml
def main():
logger.warning('This script is aimed to demonstrate how to convert the'
'JSON file to a single image dataset, and not to handle'
'multiple JSON f