Update channel_functions.py

channel_functions.py

@@ -112,8 +112,6 @@ def which_channel_dominates(img_arr, original_image_plot = 'yes', original_image
 
     print(np.max(red_channel), np.max(green_channel), np.max(blue_channel))
 
-    if original_image_plot == 'yes':
-        plt.imshow(cv2.cvtColor(img_arr,cv2.COLOR_BGR2RGB), alpha=original_image_opacity)
 
     which_channel_dominates = np.zeros((img_arr.shape[0],img_arr.shape[1]))
 
@@ -135,8 +133,13 @@ def which_channel_dominates(img_arr, original_image_plot = 'yes', original_image
 
     #Map the color code to the image
     fig, ax = plt.subplots(figsize=(8,8))
+
+    if original_image_plot == 'yes':
+       plt.imshow(cv2.cvtColor(img_arr,cv2.COLOR_BGR2RGB), alpha=original_image_opacity)
+
     plot = plt.imshow(which_channel_dominates, cmap=cmap, alpha=channel_opacity)
 
+    
     #Customize the ticks of the colorbar
     plt.colorbar(plot, orientation='vertical', 
                 ticks = [],