Skip to content

GitLab

"examples/research_projects/vscode:/vscode.git/clone" did not exist on "f3fbf9bfc0c4613e93faa4500629f77fae32c3e6"
Commit 75d2c47d authored by LDOUBLEV's avatar LDOUBLEV
Browse files

support multi-line text display

parent 8a6f1062
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 1 deletion
tools/infer/predict_rec.py
View file @ 75d2c47d
...@@ -117,7 +117,6 @@ if __name__ == "__main__": ...@@ -117,7 +117,6 @@ if __name__ == "__main__":
valid_image_file_list.append(image_file) valid_image_file_list.append(image_file)
img_list.append(img) img_list.append(img)
rec_res, predict_time = text_recognizer(img_list) rec_res, predict_time = text_recognizer(img_list)
rec_res, predict_time = text_recognizer(img_list)
for ino in range(len(img_list)): for ino in range(len(img_list)):
print("Predicts of %s:%s" % (valid_image_file_list[ino], rec_res[ino])) print("Predicts of %s:%s" % (valid_image_file_list[ino], rec_res[ino]))
print("Total predict time for %d images:%.3f" % print("Total predict time for %d images:%.3f" %
......
tools/infer/utility.py
View file @ 75d2c47d
...@@ -176,6 +176,12 @@ def draw_ocr(image, boxes, txts, scores, draw_txt=True, drop_score=0.5): ...@@ -176,6 +176,12 @@ def draw_ocr(image, boxes, txts, scores, draw_txt=True, drop_score=0.5):
"./doc/simfang.ttf", font_size, encoding="utf-8") "./doc/simfang.ttf", font_size, encoding="utf-8")
new_txt = str(count) + ': ' + txt + ' ' + '%.3f' % ( new_txt = str(count) + ': ' + txt + ' ' + '%.3f' % (
scores[count]) scores[count])
while len(new_txt) > 25:
tmp = new_txt
new_txt = tmp[:25]
draw_txt.text(
(20, gap * (count + 1)), new_txt, txt_color, font=font)
new_txt = tmp[25:]
draw_txt.text( draw_txt.text(
(20, gap * (count + 1)), new_txt, txt_color, font=font) (20, gap * (count + 1)), new_txt, txt_color, font=font)
count += 1 count += 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment