update get_img_file_list

ppocr/data/rec/dataset_traversal.py

@@ -22,7 +22,7 @@ import string
 import lmdb
 
 from ppocr.utils.utility import initial_logger
-from tools.infer.utility import get_image_file_list
+from ppocr.utils.utility import get_image_file_list
 logger = initial_logger()
 
 from .img_tools import process_image, get_img_data
@@ -173,6 +173,7 @@ class SimpleReader(object):
                         img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
                     norm_img = process_image(img, self.image_shape)
                     yield norm_img
+            else:
                 with open(self.label_file_path, "rb") as fin:
                     label_infor_list = fin.readlines()
                 img_num = len(label_infor_list)

tools/infer_rec.py

@@ -46,7 +46,7 @@ from ppocr.data.reader_main import reader_main
 from ppocr.utils.save_load import init_model
 from ppocr.utils.character import CharacterOps
 from ppocr.utils.utility import create_module
-from tools.infer.utility import get_image_file_list
+from ppocr.utils.utility import get_image_file_list
 logger = initial_logger()