snapshot...

modules/util/viirs_l1b_l2.py

@@ -132,12 +132,12 @@ def run_all(directory, out_directory, day_night='ANY', start=10):
                 data_h5f.close()
                 continue
             kept_cnt += kept
-            print(data_f, kept_cnt, int(100 * (kept/total)))
+            print(data_f, kept, kept_cnt)
             f_cnt += 1
 
             data_h5f.close()
 
-            if len(data_train_tiles) == 0:
+            if len(data_train_tiles) == 0 and len(data_valid_tiles) == 0:
                 continue
 
             if (f_cnt % 5) == 0:
@@ -147,9 +147,11 @@ def run_all(directory, out_directory, day_night='ANY', start=10):
                     np.save(out_directory + 'data_valid_' + str(cnt), data_valid)
                     num_valid_samples = data_valid.shape[0]
 
-                data_train = np.stack(data_train_tiles)
-                np.save(out_directory+'data_train_' + str(cnt), data_train)
-                num_train_samples = data_train.shape[0]
+                num_train_samples = 0
+                if len(data_train_tiles) > 0:
+                    data_train = np.stack(data_train_tiles)
+                    np.save(out_directory+'data_train_' + str(cnt), data_train)
+                    num_train_samples = data_train.shape[0]
 
                 data_valid_tiles = []
                 data_train_tiles = []