Minor fix to download STEGO pre-trained weights

leggedrobotics · Apr 12, 2024 · 6f8e603 · 6f8e603
1 parent 0fe93c7
commit 6f8e603
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -105,6 +105,7 @@ First clone the WVN and our STEGO reimplementation.
 mkdir ~/git && cd ~/git 
 git clone git@github.com:leggedrobotics/wild_visual_navigation.git
 git clone git@github.com:leggedrobotics/self_supervised_segmentation.git
+./self_supervised_segmentation/models/download_pretrained.sh
 ```
 
 (Recommended) Create new virtual environment.

diff --git a/docker/first_run.sh b/docker/first_run.sh
@@ -6,6 +6,9 @@ pip3 install -e /root/catkin_ws/src/self_supervised_segmentation > /dev/null
 echo "pip3 install -e /root/catkin_ws/src/wild_visual_navigation ..."
 pip3 install -e /root/catkin_ws/src/wild_visual_navigation > /dev/null
 
+echo "downloading STEGO pretrained weights to self_supervised_segmentation/models/"
+./src/self_supervised_segmentation/models/download_pretrained.sh
+
 echo "catkin build ..."
 catkin build > /dev/null