[update] main inference script

2024-11-19 22:30:23 -08:00
parent c17e4cecc0
commit 47ebf4528b
4 changed files with 188 additions and 5 deletions
--- a/sam2/sam2/modeling/sam2_base.py
+++ b/sam2/sam2/modeling/sam2_base.py
@@ -217,8 +217,7 @@ class SAM2Base(torch.nn.Module):
        self.memory_bank_obj_score_threshold = memory_bank_obj_score_threshold
        self.memory_bank_kf_score_threshold = memory_bank_kf_score_threshold

-        print(f"SAMURAI mode: {self.samurai_mode}")
-        print(f"Stable frames threshold: {self.stable_frames_threshold}")
+        print(f"\033[93mSAMURAI mode: {self.samurai_mode}\033[0m")

        # Model compilation
        if compile_image_encoder: