adjust txt word in line

PaddlePaddle · May 25, 2020 · eec9155 · eec9155
1 parent afecc49
commit eec9155
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tools/infer/utility.py b/tools/infer/utility.py
@@ -176,12 +176,14 @@ def draw_ocr(image, boxes, txts, scores, draw_txt=True, drop_score=0.5):
  "./doc/simfang.ttf", font_size, encoding="utf-8")
  new_txt = str(count) + ': ' + txt + ' ' + '%.3f' % (
  scores[count])
- while len(new_txt) > 25:
+ while len(new_txt) > 28:
  tmp = new_txt
- new_txt = tmp[:25]
+ new_txt = tmp[:28]
  draw_txt.text(
  (20, gap * (count + 1)), new_txt, txt_color, font=font)
- new_txt = tmp[25:]
+ new_txt = tmp[28:]
+ count += 1
+
  draw_txt.text(
  (20, gap * (count + 1)), new_txt, txt_color, font=font)
  count += 1