minor

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 = []