Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ def query_image(img, text_queries, score_threshold):
|
|
| 28 |
|
| 29 |
outputs.logits = outputs.logits.cpu()
|
| 30 |
outputs.pred_boxes = outputs.pred_boxes.cpu()
|
| 31 |
-
results = processor.
|
| 32 |
boxes, scores, labels = results[0]["boxes"], results[0]["scores"], results[0]["labels"]
|
| 33 |
|
| 34 |
font = cv2.FONT_HERSHEY_SIMPLEX
|
|
|
|
| 28 |
|
| 29 |
outputs.logits = outputs.logits.cpu()
|
| 30 |
outputs.pred_boxes = outputs.pred_boxes.cpu()
|
| 31 |
+
results = processor.post_process_object_detection(outputs=outputs, target_sizes=target_sizes, threshold=0)
|
| 32 |
boxes, scores, labels = results[0]["boxes"], results[0]["scores"], results[0]["labels"]
|
| 33 |
|
| 34 |
font = cv2.FONT_HERSHEY_SIMPLEX
|