未验证 提交 910c6128 编写于 作者: D Double_V 提交者: GitHub

Merge pull request #1203 from LDOUBLEV/fixocr

fix typo and print
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from ppocr.utils.utility import enable_static_mode
import os import os
import sys import sys
...@@ -43,7 +44,6 @@ logger = initial_logger() ...@@ -43,7 +44,6 @@ logger = initial_logger()
from ppocr.utils.save_load import init_model from ppocr.utils.save_load import init_model
def main(): def main():
startup_prog, eval_program, place, config, _ = program.preprocess() startup_prog, eval_program, place, config, _ = program.preprocess()
...@@ -72,4 +72,5 @@ def main(): ...@@ -72,4 +72,5 @@ def main():
if __name__ == '__main__': if __name__ == '__main__':
enable_static_mode()
main() main()
...@@ -191,10 +191,13 @@ if __name__ == "__main__": ...@@ -191,10 +191,13 @@ if __name__ == "__main__":
if count > 0: if count > 0:
total_time += elapse total_time += elapse
count += 1 count += 1
print("Predict time of %s:" % image_file, elapse) logger.info("The predicted time of img: {} is {}:".format(image_file,
elapse))
src_im = utility.draw_text_det_res(dt_boxes, image_file) src_im = utility.draw_text_det_res(dt_boxes, image_file)
img_name_pure = os.path.split(image_file)[-1] # image_file.split("/")[-1] img_name_pure = image_file.split("/")[-1]
img_path = os.path.join(draw_img_save, "det_res_%s" % img_name_pure) cv2.imwrite(
cv2.imwrite(img_path, src_im) os.path.join(draw_img_save, "det_res_%s" % img_name_pure), src_im)
logger.info("The visualized img saved in {}".format(
os.path.join(draw_img_save, "det_res_%s" % img_name_pure)))
if count > 1: if count > 1:
print("Avg Time:", total_time / (count - 1)) logger.info("Avg Time:", total_time / (count - 1))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册