Update channel_functions.py

channel_functions.py

@@ -105,7 +105,7 @@ def which_channel_dominates(img_arr, original_image_plot = 'yes', original_image
     print(np.unique(which_channel_dominates))
 
     #Map the color code to the image
-    # fig, ax = plt.subplots(figsize=(8,8))
+    fig, ax = plt.subplots(figsize=(8,8))
     plot = plt.imshow(which_channel_dominates, cmap=cmap, alpha=channel_opacity)
 
     #Customize the ticks of the colorbar
@@ -117,4 +117,7 @@ def which_channel_dominates(img_arr, original_image_plot = 'yes', original_image
     
     text = "Which channel dominates in the image below?\nWhite : R=G=B=255, Black : R=G=B=0\nGray : 0 < R=G=B< 255"
     plt.title(text)
-    plt.plot()
+    fig.canvas.draw()
+    # plt.plot()
+    image_array = np.array(fig.canvas.renderer.buffer_rgba())
+    return image_array