colourise_image¶

colourise_image(img, cmap='viridis')[source]¶

If input img is single channel, colourise it.

Parameters:

  • img (ndarray) – Single channel or RGB image as ndarray.

  • cmap (str) – Colormap to use, must be a valid matplotlib cmap string.

Returns:

An RGB image.

Return type:

img(ndarray)