From 633ebfd779747076bb565ca4206a4a6f4623cddf Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 1 Jun 2021 15:49:24 -0500
Subject: [PATCH] minor
---
modules/util/plot.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
diff --git a/modules/util/plot.py b/modules/util/plot.py
index 9bc6bc5c..66b33daa 100644
--- a/modules/util/plot.py
+++ b/modules/util/plot.py
@@ -10,6 +10,7 @@ from sklearn.metrics import roc_auc_score, roc_curve
import tensorflow as tf
import matplotlib
import matplotlib.pyplot as plt
+import cartopy.crs as ccrs
import os
from util.setup import home_dir
@@ -180,3 +181,19 @@ def plot_image(image, fname, label, cmap='Greys'):
Fig.clf() # Otherwise fig stays open every time
plt.close() # Close current figure
# plt.close('all')
+
+
+def plot_image2(image, cmap='Greys'):
+ fig = plt.figure(figsize=(12, 8))
+ ax = plt.axes(projection=ccrs.PlateCarree())
+ img_extent = [-180, 180, -90, 90]
+ ax.imshow(image, origin='lower', extent=img_extent, transform=ccrs.PlateCarree(), cmap=cmap, vmin=-1.5, vmax=2.5)
+ ax.coastlines(resolution='50m', color='blue', linewidth=1)
+
+ # Save to file
+ ImageDirAndName = os.path.join('/Users/tomrink', 'testimage')
+ fig.savefig(ImageDirAndName)
+
+ # Wrap up
+ fig.clf() # Otherwise fig stays open every time
+ plt.close() # Close current figure
\ No newline at end of file
--
GitLab