diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 0b82440f9c875a1b0e9f964f76711c979f36eb4c..7dc46789b37b1f209590038b4356c38f6d5d3803 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])