欢迎光临散文网 会员登陆 & 注册

深度图与置信度图的可视化(转换为伪彩色图)

2023-01-03 18:49 作者:B站自律版用户  | 我要投稿

一、深度图的可视化

以上代码解释:

1.首先得有深度图这个二维数组变量depth(一般是深度估计的算法或者AI模型可以得到)

2.然后读取深度图二维数组的最小深度与最大深度,再归一化到 [0,1] ,然后乘255映射到0~255之间,astype把二维数组转为np.uint8格式后,才能使用opencv的伪彩图函数处理为可视化的伪彩色图,最后使用CV2.imwrite()把图存到了指定的目录。

注:需要展示的可以使用matlibplot或者opencv的imshow展示,这里就不放代码了,也可以直接使用图片浏览器打开存储在指定目录的depth_img。

二、置信度图(概率图)的可视化

以上代码解释:

1.由于概率图的二维数组里的数值都是0-1之间的概率,因此不需要进行归一化

2.直接进行0~255之间的映射,然后使用CV2.applyColorMap进行伪彩图转换即可。

3.最后依旧是使用CV2.imwrite()把转换后的伪彩图proba_img保存到本地的指定目录。

注:opencv有多种伪彩图的模式,参数及对应色彩模式如下图

伪彩图参数 - 对应色彩模式


深度图与置信度图的可视化(转换为伪彩色图)的评论 (共 条)

分享到微博请遵守国家法律