1. from skimage.exposure import rescale_intensity
  2. import matplotlib.pyplot as plt
  4. def gray2color(u,channel):
  5.     """
  6.     Compute color image from intensity in fluorescence in a given channel.
  7.     Arguments:
  8.     -----------
  9.         u: np.ndarray
  10.             Input fluorescence image (2D).
  11.         channel: int
  12.             Channel to code the image in (0: Red, 1: Green, 2: Blue).
  13.     Returns:
  14.     -----------
  15.         u_color: np.ndarray
  16.             The computed output image in color.
  17.     """
  18.     u_color = np.dstack((
  19.         rescale_intensity(u if channel==0 else np.zeros_like(u), out_range='float'),
  20.         rescale_intensity(u if channel==1 else np.zeros_like(u), out_range='float'),
  21.         rescale_intensity(u if channel==2 else np.zeros_like(u), out_range='float'),
  22.         ))
  23.     return u_color
  25. def show_Green(img):
  26.     plt.imshow(gray2color(img,1))
  27.     plt.show()
  30. plt.figure()
  31. show_Green(inputImage)

Reference

https://www.jscholler.com/2020-04-20-Outer_segment_orientation/