Spaces:
Running
Running
Update cord_inference.py
Browse files- 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
|
60 |
-
|
61 |
-
|
62 |
-
|
|
|
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):
|