实例分割：label文件转mask

### 将Label文件转换为Mask用于实例分割

为了将标签文件转换为适用于实例分割的mask，通常涉及几个关键步骤。这些过程可以概括为读取原始标签数据、解析并处理该数据以创建二值化或彩色编码的mask图像，最后保存这些mask以便进一步使用。

#### 使用Python脚本自动化此流程

对于来自`labelme`格式JSON文件的情况，可以通过编写专门设计的Python程序来简化这个转变过程：

import json
from PIL import Image, ImageDraw
import numpy as np
import os


def create_mask_from_labelme(json_file_path, output_folder):
    with open(json_file_path) as f:
        data = json.load(f)

    img_width = int(data['imageWidth'])
    img_height = int(data['imageHeight'])

    mask_img = Image.new('L', (img_width, img_height), 0)
    
    draw = ImageDraw.Draw(mask_img)

    shapes = [(shape["points"], shape["label"]) for shape in data["shapes"]]
    
    unique_labels = set([s[1] for s in shapes])
    
    color_map = {label: i+1 for i, label in enumerate(unique_labels)}

    for points, label in shapes:
        polygon = tuple(tuple(point) for point in points)
        
        fill_color = color_map[label]

        draw.polygon(polygon, outline=fill_color, fill=fill_color)

    base_name = os.path.basename(os.path.splitext(json_file_path)[0])

    save_path = os.path.join(output_folder, '{}_mask.png'.format(base_name))

    mask_img.save(save_path)


if __name__ == '__main__':
    input_json_directory = './labelme_annos/'
    output_masks_directory = './masks/'

    if not os.path.exists(output_masks_directory):
        os.makedirs(output_masks_directory)

    files = [f for f in os.listdir(input_json_directory) if f.endswith('.json')]

    for file in files:
        full_path = os.path.join(input_json_directory, file)
        create_mask_from_labelme(full_path, output_masks_directory)

上述代码片段展示了如何遍历指定目录中的所有`.json`文件，并基于其中的信息生成相应的mask图像[^2]。每张mask图像是根据对应的标注形状绘制而成，不同类别的对象被赋予不同的灰度级表示，在实际应用中可以根据需求调整颜色映射逻辑。

此外，当涉及到从其他类型的标签文件（如YOLO格式）转换时，则需相应修改解析部分以适应特定于那种格式的数据结构和字段名称。