mirrors
/
GroundingDINO
mirror of https://github.com/IDEA-Research/GroundingDINO.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #87 from darshats/main 

create "." separated caption
pull/89/head
Ren Tianhe 2023-05-04 01:56:40 +08:00 committed by GitHub
parent a4dcf5d411 168d65d5c4
commit db4e6d9680
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
groundingdino/util/inference.py
View File

@ -189,7 +189,7 @@ class Model:
box_annotator = sv.BoxAnnotator() box_annotator = sv.BoxAnnotator()
annotated_image = box_annotator.annotate(scene=image, detections=detections) annotated_image = box_annotator.annotate(scene=image, detections=detections)
""" """
caption = ", ".join(classes) caption = ". ".join(classes)
processed_image = Model.preprocess_image(image_bgr=image).to(self.device) processed_image = Model.preprocess_image(image_bgr=image).to(self.device)
boxes, logits, phrases = predict( boxes, logits, phrases = predict(
model=self.model, model=self.model,