support gdino local model (load local ckpt)

2024-08-01 17:58:42 +08:00
parent 1dacb47840
commit f5b99eea3d
3 changed files with 135 additions and 1 deletions
--- a/grounded_sam2_hf_model_demo.py
+++ b/grounded_sam2_hf_model_demo.py
@@ -69,8 +69,8 @@ input_boxes = results[0]["boxes"].cpu().numpy()

 masks, scores, logits = sam2_predictor.predict(
    point_coords=None,
-    box=input_boxes,
    point_labels=None,
+    box=input_boxes,
    multimask_output=False,
 )