diff --git a/modules/util/plot.py b/modules/util/plot.py index 41040f7a3f43853058d5322f6886a6faac65b991..dbfac24818da9320af84be5c4933a0d31e86f0d1 100644 --- a/modules/util/plot.py +++ b/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')