Spaces:

P3ngLiu
/

FO1_VS_SAM3_DEMO

Running on Zero

P3ngLiu commited on 7 days ago

Commit

24ae542

verified ·

1 Parent(s): 5ddd5a7

Update demo/gradio_demo2.py

Files changed (1) hide show

demo/gradio_demo2.py CHANGED Viewed

@@ -360,7 +360,7 @@ if __name__ == "__main__":
     # Load SAM3
     sam3_model = build_sam3_image_model(checkpoint_path='./sam3.pt', device="cuda",bpe_path='/home/user/app/resources/bpe_simple_vocab_16e6.txt.gz')
-    sam3_processor = Sam3Processor(sam3_model, confidence_threshold=0.0, device="cuda")
     # Load UPN
     upn_model = UPNWrapper(upn_ckpt_path)

     # Load SAM3
     sam3_model = build_sam3_image_model(checkpoint_path='./sam3.pt', device="cuda",bpe_path='/home/user/app/resources/bpe_simple_vocab_16e6.txt.gz')
+    sam3_processor = Sam3Processor(sam3_model, confidence_threshold=0.5, device="cuda")
     # Load UPN
     upn_model = UPNWrapper(upn_ckpt_path)