From f8302ee886ef9b631f11a52900dac964a61350e1 Mon Sep 17 00:00:00 2001
From: fredeee
Date: Thu, 2 Nov 2023 10:47:21 +0100
Subject: initiaö commit
---
scripts/exec/eval_savi.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
create mode 100644 scripts/exec/eval_savi.py
(limited to 'scripts/exec/eval_savi.py')
diff --git a/scripts/exec/eval_savi.py b/scripts/exec/eval_savi.py
new file mode 100644
index 0000000..87d4e59
--- /dev/null
+++ b/scripts/exec/eval_savi.py
@@ -0,0 +1,17 @@
+import argparse
+import sys
+from data.datasets.ADEPT.dataset import AdeptDataset
+from scripts.evaluation_adept_savi import evaluate
+
+if __name__ == "__main__":
+ parser = argparse.ArgumentParser()
+ parser.add_argument("-load", default="", type=str, help='path to savi slots')
+ parser.add_argument("-n", default="", type=str, help='results name')
+
+ # Load configuration
+ args = parser.parse_args(sys.argv[1:])
+ print(f'Evaluating savi slots {args.load}')
+
+ # Load dataset
+ testset = AdeptDataset("./", 'adept', 'createdown', (30 * 2**(2*2), 20 * 2**(2*2)))
+ evaluate(testset, args.load, args.n, plot_frequency= 1, plot_first_samples = 2)
\ No newline at end of file
--
cgit v1.2.3