diff --git a/configs/rec/rec_chinese_lite_train.yml b/configs/rec/rec_chinese_lite_train.yml index 16b7148..738aaa8 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 927a4ca..70e7551 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():