mp-02 commited on
Commit
b11d3ce
·
verified ·
1 Parent(s): f83550f

Update sroie_inference.py

Browse files
Files changed (1) hide show
  1. sroie_inference.py +1 -1
sroie_inference.py CHANGED
@@ -96,7 +96,7 @@ def prediction(image):
96
  #draw = ImageDraw.Draw(image, "RGBA")
97
  font = ImageFont.load_default()
98
 
99
- for prediction, box in zip(preds, bboxes):
100
  draw.rectangle(box)
101
  draw.text((box[0]+10, box[1]-10), text=prediction, font=font, fill="black", font_size="8")
102
 
 
96
  #draw = ImageDraw.Draw(image, "RGBA")
97
  font = ImageFont.load_default()
98
 
99
+ for prediction, box in zip(true_predictions, true_boxes):
100
  draw.rectangle(box)
101
  draw.text((box[0]+10, box[1]-10), text=prediction, font=font, fill="black", font_size="8")
102