浅谈语义分割任务如何给像素点上色

通常解码器最终输出都会是(batch_size ,num_classes, h, w)

num_classes对应任务的类别数

所以在预测任务中,数据经过模型得到预测结果,形状也是

(batch_size ,num_classes, h, w)的Tensor列表,元素都是预测概率

seg_pred = net(x) #

1.创建画布,设置每个类别的颜色(我这里是4个,背景不算,因为不需上色)

    img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)    lane_img = np.zeros_like(img)  # 创建画布    color = np.array([[255, 125, 0], [0, 255, 0], [0, 0, 255], [0, 255, 255]], dtype='uint8')   

2. 返回最大概率的通道数,通道1对应类别1, 类推。(开头说了,语义分割解码器n个channel对应n个类别

# 返回预测概率数组中最大值的索引-------axis=0,返回概率值大的属于哪一个通道。对应语义分割的类别。

shape是(h,w) 一共有h组列表, 每个列表有w个元素。 代表从上到下,每一行像素点的分类都给出来了

coord_mask = np.argmax(seg_pred, axis=0)

3.上色

or i in range(0, num_classes-1): #不算背景           lane_img[coord_mask == (i + 1)] = color[i]  # 给像素点上色-----例如    img = cv2.addWeighted(src1=lane_img, alpha=0.8, src2=img, beta=1., gamma=0.)
(0)

相关推荐

  • 憨批的语义分割重制版6——Pytorch 搭建自己的Unet语义分割平台

    注意事项 学习前言 什么是Unet模型 代码下载 Unet实现思路 1.训练文件详解 2.LOSS解析 1.主干网络介绍 2.加强特征提取结构 3.利用特征获得预测结果 一.预测部分 二.训练部分 训 ...

  • 【CV】基于阈值处理的图像分割算法!

    图像处理 Author:louwill Machine Learning Lab 基于阈值的图像分割因其处理直观.实现简单和计算速度快,是一种更为常用的传统图像分割算法.本文基于图像灰度阈值处理的基本 ...

  • 关联分割点云中的实例和语义

    2019 CVPR Xinlong Wang 1 Shu Liu 2 Xiaoyong Shen 2 Chunhua Shen 1 Jiaya Jia 2,3 1 The University of ...

  • 基于OpenCV的区域分割、轮廓检测和阈值处理

    重磅干货,第一时间送达 OpenCV是一个巨大的开源库,广泛用于计算机视觉,人工智能和图像处理领域.它在现实世界中的典型应用是人脸识别,物体检测,人类活动识别,物体跟踪等. 现在,假设我们只需要从整个 ...

  • 【开源方案共享】PASS3D:精确且快速的3D点云语义分割

    标题:PASS3D: Precise and Accelerated Semantic Segmentation for 3D PointCloud 作者:in Kong1, Guangyao Zha ...

  • 浅谈三维点云中的几何语义

    前言: 本文介绍了点云中不可忽视的一项重要属性--几何语义,并尝试根据自己的理解和实践经验对其进行一些归纳总结,可能有些地方有理解有误,请大家抱着批判的态度学习. 注1:文末附[三维点云]交流群 注2 ...

  • 从传统到深度学习:浅谈点云分割中的图结构

    随着3D扫描技术的进步,如何将点云的前景和背景正确分离成为点云处理的一个具有挑战性的问题.具体来说,就是给定一个对象位置的估计,目标是识别属于该对象的那些点,并将它们与背景点分开.除了将前景与背景分离 ...

  • 浅谈乡村治理模式发生了哪些变化?

    随着乡村现代化的发展,越来越多的乡村走上了建设数字乡村的道路,从传统的乡村治理到使用互联网数字化治理的模式,乡村的风貌和农民的生活也发生了巨大的变化,接下来就让我们一起来了解一下乡村治理模式到底发生了 ...

  • 汽车是怎么开发出来的?浅谈汽车开发流程

    许良  汽车话题下的优秀答主你知道汽车是怎么开发出来的吗?你的脑海中很可能浮现出来这样一个画面:一个非常有艺术气息的设计师,在草图上帅气的描绘着看起来非常犀利的线条.对,但不全对.对于汽车工程师的我而 ...

  • 浅谈办公室装修的发展前景和趋势

    未来办公室装修的发展趋势会是怎样的?这是这个行业未来前景的重要话题.在这样一个新时代里,所有的事物都会以最新颖的方式出发.科技的发展也让每个行业都转遍了方向,同时对行业的要求和品质也有了更高的要求. ...

  • 颧骨浅谈

    ​骨过高 颧骨过于发达的人,单从脸部看上去就给人以高傲的感觉.而他们也得确有这种个性,常固执已见,虚张声势或显得自负是他们的特色.在工作上,也不愿意接受他人的忠告,总认为自己的就是最好的,这一点不利于 ...

  • 浅谈地龙在治疗咳喘中的运用

    浅谈地龙在治疗咳喘中的运用 笔者从事中医临床工作十多年来,在运用地龙治疗痰湿壅肺型.肺络瘀阻型.心肺两虚型.肾不纳气型的咳喘上有了一些心得,现介绍如下.   咳喘是现代医学中呼吸系统.心血管系统疾病的 ...

  • [鉴史释疑]曹操是英雄还是汉贼 浅谈曹操的英雄之处

    时间:2021-04-16 08:30:03    来源:本站(吾爱诗经网)整理       作者:魁哥说历史 在不同人眼里,曹操的形象和品格都是不一样的.有人觉得曹操是汉贼,但也有人说曹操是英雄,其 ...

  • 油车的教学真能开电车? 从驾校角度浅谈电动汽车事故

    可能是因为电动汽车正处汽车变革的"风口"中,所以因为电动汽车造成的事故在今年都能成为热搜榜的常客,刨去产品本身的原因不谈,在驾校的培训体系中有没有需要提升的部分? 我们都知道目前驾 ...