【语义分割数据预处理】

1 Labelme 标注数据

安装labelme

pip install lamebelme

使用labelme

activate labelme
labelme

得到json文件

2 json2mask

import os
import shutil

def json_to_mask(json_path, process_path, mask_path, img_path):
    # 1) 将每一个json文件转换为dataset文件夹存入process中，文件夹名与json文件名相同
    # 文件名为 X.json
    filelist = os.listdir(json_path)
    for file in filelist:
        filename = file.split('.')[0]   # X.json分离得到X字符
        new_path = filename
        cmd = 'labelme_json_to_dataset ' + json_path + '/' + file + ' -o ' + process_path + '/' + new_path      #单双引号应该都可以
        os.system(cmd)

    # 2) 将process中每个文件夹中的label.png文件找出，重命名并保存至mask文件夹
    for folder in os.listd