Spaces:

segments
/

panoptic-segment-anything

Runtime error

Tobias Cornille commited on Apr 19, 2023

Commit

9cd2e3e

1 Parent(s): da3ec5c

Improve detection prompt

Files changed (1) hide show

app.py CHANGED Viewed

@@ -88,7 +88,7 @@ def dino_detection(
     device,
     visualize=False,
 ):
-    detection_prompt = " . ".join(category_names)
     dino_image = load_image_for_dino(image)
     dino_image = dino_image.to(device)
     with torch.no_grad():
@@ -100,6 +100,7 @@ def dino_detection(
             text_threshold=text_threshold,
             device=device,
         )
     category_ids = [category_name_to_id[phrase] for phrase in phrases]
     if visualize:

     device,
     visualize=False,
 ):
+    detection_prompt = " ;. ".join(category_names) + " ;."
     dino_image = load_image_for_dino(image)
     dino_image = dino_image.to(device)
     with torch.no_grad():
             text_threshold=text_threshold,
             device=device,
         )
+    phrases = [phrase.replace(" ;", "").strip() for phrase in phrases]
     category_ids = [category_name_to_id[phrase] for phrase in phrases]
     if visualize: