snapshot...

modules/util/plot.py

@@ -157,7 +157,7 @@ def plot_roc_curve(correct_labels, prob_pos_class):
     plt.savefig('/home/rink/test_roc', dpi=320)
 
 
-def plot_image(image, fname=None, cmap='Greys', vmin=-1.5, vmax=2.5):
+def plot_image(image, fname=None, cmap='Greys', vmin=None, vmax=None):
     Fig, ax = plt.subplots()
 
     # Plot the image
@@ -187,8 +187,8 @@ def plot_image(image, fname=None, cmap='Greys', vmin=-1.5, vmax=2.5):
         Fig.show()
 
     # Wrap up
-    Fig.clf()  # Otherwise fig stays open every time
-    plt.close()  # Close current figure
+    #Fig.clf()  # Otherwise fig stays open every time
+    #plt.close()  # Close current figure
     # plt.close('all')