update scripts

martius-lab · Mar 26, 2024 · 24c57de · 24c57de
1 parent fa80bfe
commit 24c57de
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/scripts/run_all.py b/scripts/run_all.py
@@ -12,7 +12,7 @@
 for seed in range(10):
     os.system(f"python3 {p} --experiment {exp} --seed {seed}")
 if exp in exps[:2]:
-    os.system("python3 " + str(os.path.join(HITCHHIKING_ROOT_DIR, "visu", "figure_19.py", exp)))
-    os.system("python3 " + str(os.path.join(HITCHHIKING_ROOT_DIR, "visu", "figure_12a.py", exp)))
+    os.system("python3 " + str(os.path.join(HITCHHIKING_ROOT_DIR, "visu", "figure_19.py ")+ exp))
+    os.system("python3 " + str(os.path.join(HITCHHIKING_ROOT_DIR, "visu", "figure_12a.py ") + exp))
 else:
     os.system("python3 " + str(os.path.join(HITCHHIKING_ROOT_DIR, "visu", "figure_12b.py")))
diff --git a/visu/figure_12a.py b/visu/figure_12a.py
@@ -43,11 +43,11 @@
 df = pd.DataFrame.from_dict(df_res)
 
 mapping = {
-    "r9_svd": RotRep.SVD,
-    "r6_gso": RotRep.GSO,
-    "quat_c": RotRep.QUAT_C,
-    "rotvec": RotRep.EXP,
-    "euler": RotRep.EULER,
+    "r9_svd": str(RotRep.SVD) + "-Chordal",
+    "r6_gso": str(RotRep.GSO) + "-Chordal",
+    "quat_c": str(RotRep.QUAT_C) + "-Chordal",
+    "rotvec": str(RotRep.EXP) + "-Chordal",
+    "euler":  str(RotRep.EULER) + "-Chordal",
 }
 
 for k, v in mapping.items():
@@ -73,7 +73,7 @@
     fliersize=2.5,
     showfliers=True,
 )
-plt.xlabel("Error - Geodesic Distance")
+plt.xlabel("Error - Geodesic")
 plt.ylabel("")
 plt.tight_layout()