create "." separated caption

This commit is contained in:
Darshat Shah
2023-05-03 23:23:32 +05:30
parent a4dcf5d411
commit 168d65d5c4

View File

@@ -189,7 +189,7 @@ class Model:
box_annotator = sv.BoxAnnotator()
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)
boxes, logits, phrases = predict(
model=self.model,