Update channel_functions.py

channel_functions.py

@@ -42,7 +42,25 @@ def individual_channel_image(img_arr, channel= 'r', ax=None):
       ax.set_title('Image in the {} channel'.format(channel_name))
 
 
-
+def individual_channel_image_final(img_arr, channel='Red'):
+    if channel in ['Red','Green','Blue']:
+        fig, ax = plt.subplots(figsize = (10,10))
+        individual_channel_image(img_arr, channel= channel)
+        plt.show()
+        fig.canvas.draw()
+        image_array = np.array(fig.canvas.renderer.buffer_rgba())
+        return image_array
+
+    if channel in ['All']:
+        fig, ax = plt.subplots(2,2, figsize = (12,10))
+        individual_channel_image(img, channel='r', ax=ax[0,0])
+        individual_channel_image(img, channel='g', ax=ax[0,1])
+        individual_channel_image(img, channel='b', ax=ax[1,0])
+        individual_channel_image(img, channel='full', ax=ax[1,1])
+        plt.show()
+        fig.canvas.draw()
+        image_array = np.array(fig.canvas.renderer.buffer_rgba())
+        return image_array
 
 def channel_distribution_plotter(img_array):