Update channel_functions.py

channel_functions.py

@@ -28,7 +28,7 @@ def individual_channel_image(img_arr, channel= 'r', ax=None):
       plt.imshow(cv2.cvtColor(img_arr,cv2.COLOR_BGR2RGB))
     else:
       plt.imshow(plot_arr, cmap = cmap)
-      plt.colorbar(orientation= 'vertical', shrink = 0.7, pad = 0.01, fraction = 0.035,)
+      plt.colorbar(orientation= 'vertical', shrink = 0.7, pad = 0.01, fraction = 0.046,)
       plt.title('Image in the {} channel'.format(channel_name))
       plt.show()
 
@@ -42,7 +42,7 @@ def individual_channel_image(img_arr, channel= 'r', ax=None):
       # ax.imshow(cv2.cvtColor(img_arr,cv2.COLOR_RGB2BGR))
     else:
       plot = ax.imshow(plot_arr, cmap = cmap)
-      plt.colorbar(plot, orientation= 'vertical', ax = ax)
+      plt.colorbar(plot, orientation= 'vertical', ax = ax, fraction = 0.046,)
       # plt.colorbar(orientiation= 'vertical', shrink = 0.7, pad = 0.1)
       ax.set_title('Image in the {} channel'.format(channel_name))