wendys-llc commited on
Commit
ce9f3bd
·
verified ·
1 Parent(s): 066732d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.post_process(outputs=outputs, target_sizes=target_sizes)
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