Update channel_functions.py

channel_functions.py

@@ -119,5 +119,5 @@ def which_channel_dominates(img_arr, original_image_plot = 'yes', original_image
     plt.title(text)
     fig.canvas.draw()
     # plt.plot()
-    image_array = np.array(fig.canvas.renderer.buffer_rgb())
+    image_array = np.array(fig.canvas.renderer.buffer_rgba())
     return image_array