binary segmentation 数据处理脚本小结

原创

已于 2023-05-31 15:04:35 修改 · 400 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #json #开发语言

于 2023-01-30 15:33:06 首次发布

本文介绍如何将JSON文件转换为图像及对应的mask，并实现mask的二值化处理与图像裁剪。首先通过解析JSON文件内容，利用图像处理库生成对应的图像与mask；接着将生成的mask进行二值化处理以便于后续的图像分析任务；最后，通过指定坐标对图像进行裁剪，以获取关注区域。

1.首先是json2image，废话少说，看代码

import base64
import json
import os
import os.path as osp
import uuid

import imgviz
import PIL.Image
from labelme import utils

if __name__ == "__main__":

    base_dir = osp.dirname(osp.abspath(__file__))
#输出图像与mask路径
    out_dir_name = "output"
    out_dir = osp.join(base_dir, out_dir_name)
    if not osp.exists(out_dir):
        os.mkdir(out_dir)
    label_names = []
    label_file = "./label.txt"
    label_value_dict = {
   
   }
    with open(label_file) as f:
        labels = f.readlines()
        if len(labels) > 0:
            for i, item in enumerate(labels):
                item = item.strip()
                label_value_dict[item] = i
#json输入标签
    json_file_name = "json"  #'cityspaces'
    json_file_path = osp.join(base_dir, json_file_name)
    #第一张图从几开始
    new_filename = 1095
    for file_name in os.listdir(json_file_path)