DebasishDhal99 commited on
Commit
202eafe
·
1 Parent(s): 6e5e24d

Update channel_functions.py

Browse files
Files changed (1) hide show
  1. channel_functions.py +1 -1
channel_functions.py CHANGED
@@ -27,7 +27,7 @@ def individual_channel_image(img_arr, channel= 'r', ax=None):
27
  if channel_name == 'Original':
28
  plt.imshow(cv2.cvtColor(img,cv2.COLOR_BGR2RGB))
29
  else:
30
- plt.imshow(plot_arr)
31
  plt.colorbar(orientation= 'vertical', shrink = 0.7, pad = 0.01)
32
  plt.title('Image in the {} channel'.format(channel_name))
33
  plt.show()
 
27
  if channel_name == 'Original':
28
  plt.imshow(cv2.cvtColor(img,cv2.COLOR_BGR2RGB))
29
  else:
30
+ plt.imshow(plot_arr, cmap = cmap)
31
  plt.colorbar(orientation= 'vertical', shrink = 0.7, pad = 0.01)
32
  plt.title('Image in the {} channel'.format(channel_name))
33
  plt.show()