以下代码支持,可以支持新版本的labelme
,不需要要卸载重新安装labelme 3.16.7- 可以直接根据当前新版本的labelme标注好的json文件,直接进行mask转换
- 这个写的比较细节
,其他转换都或多少不好用import argparse import base64 import json import os import os.path as osp import imgviz import PIL.Image from labelme.logger import logger from labelme import utils def main(): logger.warning( "This script is aimed to demonstrate how to convert the " "JSON file to a single image dataset." ) logger.warning( "It won't handle multiple JSON files to generate a " "real-use dataset." ) # json_file是标注完之后生成的json文件的目录。out_dir是输出目录,即数据处理完之后文件保存的路径 json_file = r"D:\img\json_dir" out_jpgs_path = "datasets/JPEGImages" out_mask_path = "datasets/SegmentationClass" # 如果输出的路径不存在,则自动创建这个路径 if not osp.exists(out_jpgs_path): os.mkdir(out_jpgs_path) if not osp.exists(out_mask_path): os.mkdir(out_mask_path) for file_name in os.listdir(json_file): # 遍历json_file里面所有的文件,并判断这个文件是不是以.json结尾 if file_name.endswith(".json"): path = os.path.join(json_file, file_name) if os.path.isfile(path): data = json.load(open(path)) # 获取json里面的图片数据,也就是二进制数据 imageData = data.get("imageData") # 如果通过data.get获取到的数据为空,就重新读取图片数据 if not imageData: imagePath = os.path.join(json_file, data["imagePath"]) with open(imagePath, "rb") as f: imageData = f.read() imageData = base64.b64encode(imageData).decode("utf-8") # 将二进制数据转变成numpy格式的数据 img = utils.img_b64_to_arr(imageData) # 将类别名称转换成数值,以便于计算 label_name_to_value = {"_background_": 0} for shape in sorted(data["shapes"], key=lambda x: x["label"]): label_name = shape["label"] if label_name in label_name_to_value: label_value = label_name_to_value[label_name] else: label_value = len(label_name_to_value) label_name_to_value[label_name] = label_value lbl, _ = utils.shapes_to_label(img.shape, data["shapes"], label_name_to_value) label_names = [None] * (max(label_name_to_value.values()) + 1) for name, value in label_name_to_value.items(): label_names[value] = name lbl_viz = imgviz.label2rgb( label=lbl, image=imgviz.asgray(img), label_names=label_names, loc="rb" ) # 将输出结果保存, PIL.Image.fromarray(img).save(osp.join(out_jpgs_path, file_name.split(".")[0]+'.jpg')) utils.lblsave(osp.join(out_mask_path, "%s.png" % file_name.split(".")[0]), lbl) print("Done") if __name__ == "__main__": main()
语义分割:标注json文件转mask
猜你喜欢
转载自blog.csdn.net/weixin_61623830/article/details/134404748
今日推荐
周排行