From 71dc2545a95891ad59f154bcb6d191c6fda5ba21 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 3 Aug 2022 14:20:35 -0500
Subject: [PATCH] minor...
---
modules/util/viirs_l1b_l2.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/modules/util/viirs_l1b_l2.py b/modules/util/viirs_l1b_l2.py
index 6cbdf315..dec7098f 100644
--- a/modules/util/viirs_l1b_l2.py
+++ b/modules/util/viirs_l1b_l2.py
@@ -31,6 +31,9 @@ def run_all(directory, out_directory):
num_train_samples, num_valid_samples = 0, 0
cnt = 10
+ total_num_train_samples = 0
+ total_num_test_samples = 0
+
for p in os.scandir(directory):
if not p.is_dir():
continue
@@ -143,6 +146,10 @@ def run_all(directory, out_directory):
num_valid_samples += data_valid.shape[0]
print('num_train_samples, num_valid_samples: ', num_train_samples, num_valid_samples)
+ total_num_train_samples += num_train_samples
+ total_num_test_samples += num_valid_samples
+
+ print('total_num_train_samples, total_num_valid_samples: ', num_train_samples, num_valid_samples)
def run(data_h5f, label_h5f, data_tiles, label_tiles, mod_tile_width=64, kernel_size=9):
--
GitLab