用伪彩图表示深度图的python实现

• 深度信息只有单个通道，如果用灰度图表示灰不拉几，看不清楚，也很不好看。所以现在的深度图很多都是用伪彩图来表示，这样很好感受深度的变化，用opencv可以实现灰度图到伪彩图的转换，用的applycolormap这个函数。

• 伪彩图的样式选择：

  
  
  
  

  -代码实现

import cv2
import numpy as np
from PIL import Image
def standardization(data):#标准化变成-1到1
    mu = np.mean(data, axis=0)
    sigma = np.std(data, axis=0)
    return (data - mu) / sigma
def normalization(data):#归一化变成0-1
    _range = np.max(data) - np.min(data)
    return (data - np.min(data)) / _range

depth_pr=np.load('testModel/depth/1.npy')
depth_pr=np.reshape(depth_pr,(depth_pr.shape[1],depth_pr.shape[2]))
depth_pr=normalization(depth_pr)*255
depth_pr=depth_pr.astype(np.uint8)
im_color=cv2.applyColorMap(depth_pr,9)
im=Image.fromarray(im_color)
im.save('0.png')

• 结果