图像分割json文件转化为mask标记

工具：labelme==3.16.5版本，高版本或出现：

AttributeError: module 'labelme.utils' has no attribute 'draw_label'  这个问题。

简单安装就可以：

pip install labelme==3.16.5

全部程序：

# -*- coding: UTF-8 -*-
import argparse
import json
import os
import os.path as osp
import base64
import warnings
import PIL.Image
import yaml
from labelme import utils
import cv2
import numpy as np
from skimage import img_as_ubyte


def main(json_file):
    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.")
    #json文件加入列表
    list_path = []
    for json_single in os.listdir(json_file):
        if json_single.endswith('.json'):
            list_path.append(json_single)


    for i in range(0, len(list_path)):
        #挨个读取json文件
        path = os.path.join(json_file, list_path[i])
        if os.path.isfile(path):
            data = json.load(open(path))
            img = utils.img_b64_to_arr(data['imageData'])
            print('imagedata: ', data['imageData'])
            print('datashape: ', data['shapes'])
            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(path).replace('.', '_')
            out_dir = osp.basename(path).split('.json')[0]
            save_file_name = out_dir
            # out_dir = osp.join(osp.dirname(path), out_dir)

            if not osp.exists(json_file + 'mask'):
                os.mkdir(json_file + 'mask')
            maskdir = json_file + 'mask'

            if not osp.exists(json_file + 'mask_viz'):
                os.mkdir(json_file + 'mask_viz')
            maskvizdir = json_file + 'mask_viz'
            out_dir1 = maskdir

            #保存
            PIL.Image.fromarray(lbl).save(out_dir1 + '/' + save_file_name + '.png')
            PIL.Image.fromarray(lbl_viz).save(maskvizdir + '/' + save_file_name +'_label_viz.png')

            with open(osp.join(out_dir1, '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_dir1, 'info.yaml'), 'w') as f:
                yaml.safe_dump(info, f, default_flow_style=False)
            print('Saved to: %s' % out_dir1)


if __name__ == '__main__':
    json_file = r"C:\Users\bbwang\Desktop\701tooth\20210523_old_Images/imgs/"
    main(json_file)

发现生成的掩码，不够清晰，可以使用Image进行增强对比度：

大致代码是：
from PIL import Image, ImageEnhance
contrast=int(number)  #设置一个对比度，取255效果很好
img=Image.open(img_path, 2)
img_en=ImageEnhance.Contrast(img)
output=img_en.enhance(contrast)
output.save(new_img_path)

此外，在分割中遇到可以根据遍历像素值来界定label，主要适用于二分类像素中，二值化图像。

label_input=Image.open(data_path_label).convert('L')
label_input=label_input.resize((224, 224))
label_before=np.asarray(label_input)
for i in range(len(label_before):
    for j in range(len(label_before[i])):
        if label_before[i][j] > 128:
            label_before[i][j]=1
        else:
            label_before[i][j]=0
label = label_fore

随手一记，担心忘记。