IDEA-Research / Grounded-Segment-Anything

Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything
https://arxiv.org/abs/2401.14159
Apache License 2.0
14.11k stars 1.31k forks source link

Small fix in grounded_sam_simple_demo.py #461

Closed zs144 closed 4 months ago

zs144 commented 4 months ago

Same problem as issue #460 . The detections object initialized from grounding_dino_model.predict_with_classes() now has 6 properties, so we need 6 arguments when retrieving them. The problem is fixed by adding the sixth argument, and we can just name it as _ since we don't need it in this demo.