def plot_deph_normalized_colormap()

in metropolis/utils/color_map.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

def plot_deph_normalized_colormap(depth_map, scale_range):
    # depth=0 is the area with no depth data, will be colorized as far away background
    depth_map[depth_map <= 0] = 255
    # the color map wil be spread within 0 - scale_range
    depth_map[depth_map > scale_range] = scale_range
    depth_map = depth_map / scale_range
    cmap = plt.cm.jet_r
    rgba = cmap(depth_map)
    rgb = rgba[:, :, 0:3] * 255.0

    return rgb