From 25f3c98c12d111b1465cdb657106dc4291ef1ed1 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 28 Apr 2021 14:13:56 -0500
Subject: [PATCH] snapshot...
---
modules/icing/pirep_goes.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index 0b82440f..7dc46789 100644
--- a/modules/icing/pirep_goes.py
+++ b/modules/icing/pirep_goes.py
@@ -1010,7 +1010,7 @@ def fov_extract(outfile='/home/rink/fovs_l1b_out.h5', train_params=l1b_ds_list,
h5f_expl.close()
-def tile_extract(outfile='/home/rink/tiles_l1b_out.h5', train_params=l1b_ds_list, ds_types=l1b_ds_types):
+def tile_extract(outfile='/home/rink/tiles_l1b_out.h5', train_params=l1b_ds_list, ds_types=l1b_ds_types, augment=False):
icing_int_s = []
ice_time_s = []
no_ice_time_s = []
@@ -1041,6 +1041,10 @@ def tile_extract(outfile='/home/rink/tiles_l1b_out.h5', train_params=l1b_ds_list
for ds_name in train_params:
dat = f[ds_name][i, 12:28, 12:28]
icing_data_dct[ds_name].append(dat)
+ if augment:
+ if icing_int[i] >= 3:
+ icing_data_dct[ds_name].append(np.fliplr(dat))
+ icing_data_dct[ds_name].append(np.flipud(dat))
icing_int_s.append(icing_int[i])
ice_time_s.append(times[i])
ice_lon_s.append(lons[i])
--
GitLab