json转mask语义分割标签

importbase64importjsonimportosimportos.pathasospimportPIL.Imagefromlabelmeimportutils# 输入输出目录json_dir=r"E:\data\钻孔摄像\core_iamge\json (1)"out_jpgs_path=r"E:\data\钻孔摄像\core_iamge\new\JPEGImages"out_mask_path=r"E:\data\钻孔摄像\core_iamge\new\SegmentationClass"os.makedirs(out_jpgs_path,exist_ok=True)os.makedirs(out_mask_path,exist_ok=True)# ---------- 第一步：收集所有 JSON 中的标签名称 ----------all_labels=set()forfile_nameinos.listdir(json_dir):ifnotfile_name.endswith(".json"):continuepath=os.path.join(json_dir,file_name)withopen(path)asf:data=json.load(f)forshapeindata["shapes"]:all_labels.add(shape["label"])# ---------- 第二步：构建全局固定的映射 ----------label_name_to_value={"_background_":0}foridx,labelinenumerate(sorted(all_labels),start=1):label_name_to_value[label]=idxprint("全局类别映射:",label_name_to_value)# ---------- 第三步：转换所有 JSON ----------forfile_nameinos.listdir(json_dir):ifnotfile_name.endswith(".json"):continuepath=os.path.join(json_dir,file_name)withopen(path)asf:data=json.load(f)# 获取图片数据imageData=data.get("imageData")ifnotimageData:imagePath=os.path.join(json_dir,data["imagePath"])withopen(imagePath,"rb")asf_img:imageData=base64.b64encode(f_img.read()).decode("utf-8")img=utils.img_b64_to_arr(imageData)# 使用全局映射生成标签矩阵lbl,_=utils.shapes_to_label(img.shape,data["shapes"],label_name_to_value)# 保存原图和标签图basename=file_name.split(".")[0]PIL.Image.fromarray(img).save(osp.join(out_jpgs_path,f"{basename}.jpg"))utils.lblsave(osp.join(out_mask_path,f"{basename}.png"),lbl)print("Done")