google
/

owlvit-base-patch16

Zero-Shot Object Detection

Model card Files Files and versions

nielsr HF Staff commited on Oct 23, 2023

Commit

d3ef005

·

1 Parent(s): c51e657

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -57,7 +57,6 @@ text = texts[i]
 boxes, scores, labels = results[i]["boxes"], results[i]["scores"], results[i]["labels"]
 # Print detected objects and rescaled box coordinates
-score_threshold = 0.1
 for box, score, label in zip(boxes, scores, labels):
     box = [round(i, 2) for i in box.tolist()]
     print(f"Detected {text[label]} with confidence {round(score.item(), 3)} at location {box}")

 boxes, scores, labels = results[i]["boxes"], results[i]["scores"], results[i]["labels"]
 # Print detected objects and rescaled box coordinates
 for box, score, label in zip(boxes, scores, labels):
     box = [round(i, 2) for i in box.tolist()]
     print(f"Detected {text[label]} with confidence {round(score.item(), 3)} at location {box}")