diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py index 6fa51e70d4353158a8d537b6b51e7df5400b0e0c..c005bfb739a7b92caf24fb0ca42739713195bac2 100755 --- a/tools/infer/predict_rec.py +++ b/tools/infer/predict_rec.py @@ -117,7 +117,6 @@ if __name__ == "__main__": valid_image_file_list.append(image_file) img_list.append(img) rec_res, predict_time = text_recognizer(img_list) - rec_res, predict_time = text_recognizer(img_list) for ino in range(len(img_list)): print("Predicts of %s:%s" % (valid_image_file_list[ino], rec_res[ino])) print("Total predict time for %d images:%.3f" % diff --git a/tools/infer/utility.py b/tools/infer/utility.py index ce7ae686270beec11d319c1c3e1239d50fb9fd5f..43e68ea5f7b4dce20fa32b98b2f125ad392e3da6 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -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") new_txt = str(count) + ': ' + txt + ' ' + '%.3f' % ( 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( (20, gap * (count + 1)), new_txt, txt_color, font=font) count += 1