mp-02 commited on
Commit
98f17a7
·
verified ·
1 Parent(s): 70bfa79

Update cord_inference.py

Browse files
Files changed (1) hide show
  1. cord_inference.py +5 -4
cord_inference.py CHANGED
@@ -56,10 +56,11 @@ def prediction(image):
56
  bboxes = []
57
 
58
  for i, j in enumerate(true_predictions):
59
- if j != 'others':
60
- preds.append(true_predictions[i])
61
- l_words.append(true_words[i])
62
- bboxes.append(true_boxes[i])
 
63
 
64
  d = {}
65
  for id, i in enumerate(preds):
 
56
  bboxes = []
57
 
58
  for i, j in enumerate(true_predictions):
59
+ if true_confidence_scores[i] < 0.9: #####################################àà
60
+ true_predictions[i] = "O"
61
+ preds.append(true_predictions[i])
62
+ l_words.append(true_words[i])
63
+ bboxes.append(true_boxes[i])
64
 
65
  d = {}
66
  for id, i in enumerate(preds):