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..04ddbcd7c326cc3816b74af0a29c70dbf5d1bc4e 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -174,8 +174,7 @@ def draw_ocr(image, boxes, txts, scores, draw_txt=True, drop_score=0.5): continue font = ImageFont.truetype( "./doc/simfang.ttf", font_size, encoding="utf-8") - new_txt = str(count) + ': ' + txt + ' ' + '%.3f' % ( - scores[count]) + new_txt = str(idx) + ': ' + txt + ' ' + '%.3f' % (scores[idx]) draw_txt.text( (20, gap * (count + 1)), new_txt, txt_color, font=font) count += 1