From d9563b0a5711d283a1c30815e737e55fd8c7608f Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 12 Sep 2022 13:48:24 -0500 Subject: [PATCH] snaphot... --- modules/util/viirs_l1b_l2.py | 37 ++++++++++++++++++------------------ 1 file changed, 18 insertions(+), 19 deletions(-) diff --git a/modules/util/viirs_l1b_l2.py b/modules/util/viirs_l1b_l2.py index dbfadb86..d113d4a1 100644 --- a/modules/util/viirs_l1b_l2.py +++ b/modules/util/viirs_l1b_l2.py @@ -49,15 +49,15 @@ emis_params = ['temp_10_4um_nom', 'temp_11_0um_nom', 'temp_12_0um_nom', 'temp_13 #refl_params = ['refl_0_47um_nom', 'refl_0_65um_nom', 'refl_0_86um_nom', 'refl_1_38um_nom', 'refl_1_60um_nom'] # data_params = refl_params + emis_params -data_params = emis_params +# data_params = emis_params -l2_params = ['cld_temp_acha', 'cld_press_acha', 'cloud_fraction'] +l2_params = ['cld_temp_acha', 'cld_press_acha', 'cld_emiss_acha', 'cloud_fraction'] label_params = l2_params -#data_params = l2_params +data_params = l2_params -def run_all(directory, out_directory): +def run_all(directory, out_directory, day_night='ANY'): cnt = 10 total_num_train_samples = 0 @@ -105,8 +105,8 @@ def run_all(directory, out_directory): label_tiles = [] try: - run(data_h5f, data_params, data_tiles, tile_width=128, kernel_size=7) - run(data_h5f, label_params, label_tiles, tile_width=128, kernel_size=7) + run(data_h5f, data_params, data_tiles, tile_width=128, kernel_size=7, day_night=day_night) + # run(data_h5f, label_params, label_tiles, tile_width=128, kernel_size=7, day_night=day_night) except Exception as e: print(e) data_h5f.close() @@ -116,21 +116,20 @@ def run_all(directory, out_directory): data_h5f.close() #label_h5f.close() - if len(data_tiles) == 0 or len(label_tiles) == 0: - continue + # if len(data_tiles) == 0 or len(label_tiles) == 0: + # continue + # if len(data_tiles) != len(label_tiles): + # print('weirdness: ', data_f) + # continue - if len(data_tiles) != len(label_tiles): - print('weirdness: ', data_f) + if len(data_tiles) == 0: continue - # if len(data_tiles) == 0: - # continue - num = len(data_tiles) n_vld = int(num * 0.1) - [label_valid_tiles.append(label_tiles[k]) for k in range(n_vld)] - [label_train_tiles.append(label_tiles[k]) for k in range(n_vld, num)] + # [label_valid_tiles.append(label_tiles[k]) for k in range(n_vld)] + # [label_train_tiles.append(label_tiles[k]) for k in range(n_vld, num)] [data_valid_tiles.append(data_tiles[k]) for k in range(n_vld)] [data_train_tiles.append(data_tiles[k]) for k in range(n_vld, num)] @@ -138,15 +137,15 @@ def run_all(directory, out_directory): if f_cnt == 10: f_cnt = 0 - label_valid = np.stack(label_valid_tiles) - label_train = np.stack(label_train_tiles) + # label_valid = np.stack(label_valid_tiles) + # label_train = np.stack(label_train_tiles) data_valid = np.stack(data_valid_tiles) data_train = np.stack(data_train_tiles) np.save(out_directory+'data_train_' + str(cnt), data_train) np.save(out_directory+'data_valid_' + str(cnt), data_valid) - np.save(out_directory+'label_train_' + str(cnt), label_train) - np.save(out_directory+'label_valid_' + str(cnt), label_valid) + # np.save(out_directory+'label_train_' + str(cnt), label_train) + # np.save(out_directory+'label_valid_' + str(cnt), label_valid) label_valid_tiles = [] label_train_tiles = [] -- GitLab