labelme转换特定标签mask

liangzhengjie

已于 2022-04-12 18:11:09 修改

阅读量851

点赞数 1

CC 4.0 BY-SA版权

文章标签： python

于 2022-04-12 18:06:41 首次发布

本文链接：https://blog.csdn.net/qq_37381609/article/details/117446759

将labelme的特定标签转换成特定的mask

import json
import os
import os.path as osp
import cv2
from labelme import utils
import base64


def transJson(json_dir, spc_label=[], not_include=[], out_dir=None):
    json_file = json_dir
    if out_dir is None:
        out_dir = osp.basename(json_file).replace('.', '_')
        out_dir = osp.join(osp.dirname(json_file), out_dir)
    else:
        out_dir = out_dir
    if not osp.exists(out_dir):
        os.mkdir(out_dir)

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