Merge pull request #108 from yyyyyt123/main

YYT: fix typos in vlpart examples
Skylark0924 · Feb 2, 2024 · 14c45cb · 14c45cb
2 parents d1bb1d5 + 33cece0
commit 14c45cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rofunc/utils/visualab/segment/vlpart_sam_seg.py b/rofunc/utils/visualab/segment/vlpart_sam_seg.py
@@ -25,8 +25,8 @@ def vlpart_sam_predict(image,
     sam_ckpt_path = os.path.join(get_rofunc_path(), "learning/pre_trained_models", sam_checkpoint)
     if not os.path.exists(vlpart_ckpt_path):
         download_ckpt(
-            f"https://github.com/Cheems-Seminar/grounded-segment-any-parts/releases/download/v1.0/{vlpart_ckpt_path}",
-            vlpart_ckpt_path)
+            f"https://github.com/Cheems-Seminar/grounded-segment-any-parts/releases/download/v1.0/{vlpart_checkpoint}",
+            vlpart_checkpoint)
     if not os.path.exists(sam_ckpt_path):
         download_ckpt(f"https://dl.fbaipublicfiles.com/segment_anything/{sam_checkpoint}", sam_checkpoint)