diff --git a/configs/rec/rec_chinese_lite_train.yml b/configs/rec/rec_chinese_lite_train.yml index 16b7148045044d3efcb83ddb1d215bfcce2c8af8..738aaa81a4b6ae72ce9fc9f19c1de9babcc46bcb 100755 --- a/configs/rec/rec_chinese_lite_train.yml +++ b/configs/rec/rec_chinese_lite_train.yml @@ -15,7 +15,7 @@ Global: character_dict_path: ./ppocr/utils/ppocr_keys_v1.txt loss_type: ctc reader_yml: ./configs/rec/rec_chinese_reader.yml - pretrain_weights: best_accuracy + pretrain_weights: infer_img: ./infer_img Architecture: diff --git a/tools/infer_rec.py b/tools/infer_rec.py index 927a4ca608ec2c479505e1ea8887b899da607f59..70e75517fef7b1016d460c39f33adc689f77c6a3 100755 --- a/tools/infer_rec.py +++ b/tools/infer_rec.py @@ -20,7 +20,6 @@ import os import time import multiprocessing import numpy as np -import glob def set_paddle_flags(**kwargs): for key, value in kwargs.items():