语义分割学习笔记

语义分割学习笔记 mini菜 2023-06-03 14:57:15 629

文章目录

一、区别

  1. 语义分割:每个像素都打上标签(这个像素是人,树,背景等)语义分割只区分类别,不区分类别中的具体单元

实例分割:不光要区分类别,还要区分类别中的每一个个体

全景分割:相当于语义分割加上实例分割

二、代码演示原图像与mask融合

from PIL import Image

import imgviz

import numpy as np

image_file=r'D:\aaa\envs\labelme\Scripts\2_3_json\img.png'

mask_file=r'D:\aaa\envs\labelme\Scripts\2_3_json\label.png'

image=Image.open(image_file)

mask=Image.open(mask_file)

mask_img=Image.blend(image.convert("RGBA"),
                     mask.convert("RGBA"),0.5)

mask_img.save("vis2.png")

三、数据处理

将labelme标注好的json转换成mask图像

import json
import os

import imgviz
import numpy as np
from PIL import Image
import cv2
import glob

def save_colored_mask(mask,image_file):
    lbl_image=Image.fromarray(mask.astype(np.uint8),mode='P')
    colormap=imgviz.label_colormap()
    lbl_image.putpalette(colormap.flatten())
    lbl_image.save(image_file)

json_files=r'E:\桌面\资料\语义分割'
img_file=r'E:\桌面\资料\语义分割\图片'

json_l=glob.glob(os.path.join(json_files,'*.json'))

for json_ in json_l:
    name=os.path.basename(json_)
    img_name=name.replace('json','png')
    fs=open(json_,encoding='utf-8')
    dict_=json.load(fs)
    # 获取图像 宽,高
    height = dict_['imageHeight']
    width = dict_['imageWidth']

    shapes = dict_["shapes"]
    # 生成一个全零图像
    img = np.zeros((height, width), dtype=np.uint8)

    label_color = {"sheep": 1}
    for shape in shapes:
        # 解析多边形轮廓点的坐标
        points = shape['points']
        # 解析多边形的标签
        label = shape['label']

        points = np.array(points, dtype=np.int32)
        # 绘制轮廓
        cv2.polylines(img, [points], isClosed=True, color=(255), thickness=2)
        # 填充多边形颜色
        cv2.fillPoly(img, [points], color=label_color[label])
    img_path=os.path.join(img_file,img_name)
    print(img_path)

    save_colored_mask(img, img_path)
声明:本文内容由易百纳平台入驻作者撰写,文章观点仅代表作者本人,不代表易百纳立场。如有内容侵权或者其他问题,请联系本站进行删除。
mini菜
红包 点赞 收藏 评论 打赏
评论
0个
内容存在敏感词
手气红包
    易百纳技术社区暂无数据
相关专栏
置顶时间设置
结束时间
删除原因
  • 广告/SPAM
  • 恶意灌水
  • 违规内容
  • 文不对题
  • 重复发帖
打赏作者
易百纳技术社区
mini菜
您的支持将鼓励我继续创作!
打赏金额:
¥1易百纳技术社区
¥5易百纳技术社区
¥10易百纳技术社区
¥50易百纳技术社区
¥100易百纳技术社区
支付方式:
微信支付
支付宝支付
易百纳技术社区微信支付
易百纳技术社区
打赏成功!

感谢您的打赏,如若您也想被打赏,可前往 发表专栏 哦~

举报反馈

举报类型

  • 内容涉黄/赌/毒
  • 内容侵权/抄袭
  • 政治相关
  • 涉嫌广告
  • 侮辱谩骂
  • 其他

详细说明

审核成功

发布时间设置
发布时间:
是否关联周任务-专栏模块

审核失败

失败原因
备注
拼手气红包 红包规则
祝福语
恭喜发财,大吉大利!
红包金额
红包最小金额不能低于5元
红包数量
红包数量范围10~50个
余额支付
当前余额:
可前往问答、专栏板块获取收益 去获取
取 消 确 定

小包子的红包

恭喜发财,大吉大利

已领取20/40,共1.6元 红包规则

    易百纳技术社区