mp-02 commited on
Commit
cbee1cf
1 Parent(s): ff7a14a

Update sroie_inference.py

Browse files
Files changed (1) hide show
  1. sroie_inference.py +1 -1
sroie_inference.py CHANGED
@@ -97,7 +97,7 @@ def prediction(image):
97
  k = k[2:]
98
 
99
  blur_boxes = []
100
- for prediction, box in zip(preds, bboxes):
101
  if prediction != 'O' and prediction[2:] != 'TOTAL':
102
  blur_boxes.append(box)
103
 
 
97
  k = k[2:]
98
 
99
  blur_boxes = []
100
+ for prediction, box in zip(true_predictions, true_boxes):
101
  if prediction != 'O' and prediction[2:] != 'TOTAL':
102
  blur_boxes.append(box)
103