Update channel_functions.py

channel_functions.py

@@ -53,6 +53,7 @@ def individual_channel_image_final(img_arr, channel='Red'):
     if channel in ['Red','Green','Blue']:
         fig, ax = plt.subplots(figsize = (15,10))
         individual_channel_image(img_arr, channel= channel)
+        plt.tight_layout()
         plt.show()
         fig.canvas.draw()
         image_array = np.array(fig.canvas.renderer.buffer_rgba())
@@ -64,10 +65,11 @@ def individual_channel_image_final(img_arr, channel='Red'):
         individual_channel_image(img_arr, channel='g', ax=ax[0,1])
         individual_channel_image(img_arr, channel='b', ax=ax[1,0])
         individual_channel_image(img_arr, channel='full', ax=ax[1,1])
+        plt.tight_layout()
         plt.show()
         fig.canvas.draw()
         image_array = np.array(fig.canvas.renderer.buffer_rgba())
-        plt.close(fig)
+        # plt.close(fig)
         return image_array
 
 def channel_distribution_plotter(img_array):