labelme批量json_to_dataset转换并解决无生成yaml文件问题

关键代码

# 最前面加入导包
import yaml

# 中间是代码的主体部分就不贴了
# 然后在最下面main函数之前加上这一部分：
    logger.warning('info.yaml is being replaced by label_names.txt')
    info = dict(label_names=label_names)
    with open(osp.join(out_dir, 'info.yaml'), 'w') as f:
        yaml.safe_dump(info, f, default_flow_style=False)
    logger.info('Saved to: {}'.format(out_dir))

# 下面就是main函数了
if __name__ == '__main__':
    main()

count = os.listdir(json_dir)
    for i in range(0, len(count)):
        path = os.path.join(json_dir, count[i])

        if os.path.isfile(path):

完整代码
把整个文件代码替换就行

然后执行

labelme_json_to_dataset dir(你自己的文件夹路径)

import argparse
import base64
import json
import os
import os