initiaö commit

1 files changed, 17 insertions, 0 deletions

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