forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #1203 from LDOUBLEV/fixocr 

fix typo and print
This commit is contained in:
Double_V 2020-12-02 20:22:19 +08:00 committed by GitHub
parent f501be3c68 f2c35f55a2
commit 910c6128bb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/export_model.py
View File

@ -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()
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():
if __name__ == '__main__': if __name__ == '__main__':
enable_static_mode()
main() main()

13
tools/infer/predict_det.py
View File

@ -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))