From f9cf3f331f66422c440c4284a0bce87905a10ae0 Mon Sep 17 00:00:00 2001
From: Remi Cresson <remi.cresson@irstea.fr>
Date: Wed, 23 Feb 2022 11:14:36 +0100
Subject: [PATCH] ADD: pickle serialization
---
test/drs_import.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/test/drs_import.py b/test/drs_import.py
index 9aec815..2be81e5 100644
--- a/test/drs_import.py
+++ b/test/drs_import.py
@@ -3,12 +3,14 @@ from scenes import drs
# Arguments
parser = argparse.ArgumentParser(description="Test",)
-parser.add_argument("--root_dir", help="Root directory containing MS and PAN folders", required=True)
+parser.add_argument("--root_dir", nargs='+', help="Root directories containing MS and PAN folders", required=True)
parser.add_argument("--out_pickle", help="Output pickle file", required=True)
params = parser.parse_args()
# Get all scenes in the root_dir
-scenes = drs.get_spot67_scenes(params.root_dir)
+scenes = []
+for root_dir in params.root_dir:
+ scenes += drs.get_spot67_scenes(root_dir)
# Save scenes in a pickle file
drs.save_scenes(scenes, params.out_pickle)
--
GitLab