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

一、深度图的可视化
以上代码解释:
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有多种伪彩图的模式,参数及对应色彩模式如下图
