snapshot...

modules/util/plot.py

@@ -157,11 +157,11 @@ def plot_roc_curve(correct_labels, prob_pos_class):
     plt.savefig('/home/rink/test_roc', dpi=320)
 
 
-def plot_image(image, fname, label, cmap='Greys'):
+def plot_image(image, fname=None, cmap='Greys', vmin=-1.5, vmax=2.5):
     Fig, ax = plt.subplots()
 
     # Plot the image
-    plt.imshow(image, interpolation="nearest", cmap=cmap, vmin=-1.5,vmax=2.5)
+    plt.imshow(image, interpolation="nearest", cmap=cmap, vmin=vmin, vmax=vmax)
     # plt.imshow(image, interpolation="nearest", cmap=cmap, vmin=-0.5,vmax=1.5)
     # plt.imshow(radGrid, interpolation = "nearest", cmap = cmap, vmin = vmin, vmax = vmax)
     # plt.title(os.path.basename(InFileWV), fontsize=10)
@@ -178,11 +178,13 @@ def plot_image(image, fname, label, cmap='Greys'):
     # Finish the figure settings and save to a file
     Fig.set_size_inches([10.0, 6.6], forward=True)
     Fig.set_dpi(100)  # This seems to not get applied. Always 100?
-    # Fig.show() # Display
 
     # Save to file
-    ImageDirAndName = os.path.join('/Users/tomrink', fname)
-    Fig.savefig(ImageDirAndName)
+    if fname is not None:
+        ImageDirAndName = os.path.join('/Users/tomrink', fname)
+        Fig.savefig(ImageDirAndName)
+    else:
+        Fig.show()
 
     # Wrap up
     Fig.clf()  # Otherwise fig stays open every time