save

probcomp · Nov 5, 2023 · 173ae88 · 173ae88
1 parent 5372be8
commit 173ae88
Show file tree

Hide file tree

Showing 4 changed files with 53 additions and 99 deletions.
diff --git a/bayes3d/neural/cosypose_baseline/cosypose_setup.sh b/bayes3d/neural/cosypose_baseline/cosypose_setup.sh
@@ -12,7 +12,7 @@ else
 fi
 unset __conda_setup
 
-# # git clone --recurse-submodules https://github.com/Simple-Robotics/cosypose.git
+git clone --recurse-submodules https://github.com/Simple-Robotics/cosypose.git
 cd cosypose
 # make sure to change numpy to version 1.19.2
 sed -i 's/numpy=1.17.4/numpy=1.19.2/g' environment.yaml

diff --git a/bayes3d/viz/meshcatviz.py b/bayes3d/viz/meshcatviz.py
@@ -16,6 +16,10 @@ def setup_visualizer():
     VISUALIZER = meshcat.Visualizer()
     set_background_color([1, 1, 1])
 
+def get_visualizer():
+    global VISUALIZER
+    return VISUALIZER
+
 def set_background_color(color):
     VISUALIZER["/Background"].set_property("top_color", color)
     VISUALIZER["/Background"].set_property("bottom_color", color)