diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 43e68ea5f7b4dce20fa32b98b2f125ad392e3da6..c7bb5876f3acddff23e2a6318baaf7fa4f73ff33 100755 --- 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