diff --git a/modules/util/viirs_l1b_l2.py b/modules/util/viirs_l1b_l2.py index 23f52b41bc296ec150b464b31aa337e891bb98a7..d5c359e84e6a81c1f3368677262b466b855ec1e6 100644 --- a/modules/util/viirs_l1b_l2.py +++ b/modules/util/viirs_l1b_l2.py @@ -13,7 +13,7 @@ data_params = ['cloud_fraction'] label_params = ['cloud_fraction'] -def run_all(directory): +def run_all(directory, out_directory): cnt = 10 for p in os.scandir(directory): @@ -90,10 +90,10 @@ def run_all(directory): data_valid = np.stack(data_valid_tiles) data_train = np.stack(data_train_tiles) - np.save('/data/Personal/rink/viirs/data_train_' + str(cnt), data_train) - np.save('/data/Personal/rink/viirs/data_valid_' + str(cnt), data_valid) - np.save('/data/Personal/rink/viirs/label_train_' + str(cnt), label_train) - np.save('/data/Personal/rink/viirs/label_valid_' + str(cnt), label_valid) + 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) label_valid_tiles = [] label_train_tiles = []