Update channel_functions.py

channel_functions.py

@@ -4,7 +4,7 @@ import numpy as np
 import cv2
 
 def individual_channel_image(img_arr, channel= 'r', ax=None):
-  img_arr = np.array(img_arr[:,:,0:3], dtype=np.uint8)
+  img_arr = np.array(img_arr[:,:,0:3])
   
   if channel in ['r','red','Red']:
     plot_arr = img_arr[:,:,0]
@@ -87,8 +87,8 @@ def channel_distribution_plotter(img_array):
     plt.title("Blue Channel")
 
     plt.subplot(2,2,4)
-    # plt.imshow(cv2.cvtColor(img_array,cv2.COLOR_BGR2RGB))
-    plt.imshow(cv2.cvtColor(img_array,cv2.COLOR_RGB2BGR))    
+    plt.imshow(cv2.cvtColor(img_array,cv2.COLOR_BGR2RGB))
+    # plt.imshow(cv2.cvtColor(img_array,cv2.COLOR_RGB2BGR))    
     plt.title("Original Image")
 
     plt.suptitle("Pixel values distribution in each channel\nx-axis: pixel values, y-axis: number of pixels")