fix re bug (#7288)

tools/infer/utility.py

@@ -225,23 +225,24 @@ def create_predictor(args, mode, logger):
                     min_subgraph_size,  # skip the minmum trt subgraph
                     use_calib_mode=False)
 
-            # collect shape
-            if args.shape_info_filename is not None:
-                if not os.path.exists(args.shape_info_filename):
-                    config.collect_shape_range_info(args.shape_info_filename)
-                    logger.info(
-                        f"collect dynamic shape info into : {args.shape_info_filename}"
-                    )
+                # collect shape
+                if args.shape_info_filename is not None:
+                    if not os.path.exists(args.shape_info_filename):
+                        config.collect_shape_range_info(
+                            args.shape_info_filename)
+                        logger.info(
+                            f"collect dynamic shape info into : {args.shape_info_filename}"
+                        )
+                    else:
+                        logger.info(
+                            f"dynamic shape info file( {args.shape_info_filename} ) already exists, not need to generate again."
+                        )
+                    config.enable_tuned_tensorrt_dynamic_shape(
+                        args.shape_info_filename, True)
                 else:
                     logger.info(
-                        f"dynamic shape info file( {args.shape_info_filename} ) already exists, not need to generate again."
+                        f"when using tensorrt, dynamic shape is a suggested option, you can use '--shape_info_filename=shape.txt' for offline dygnamic shape tuning"
                     )
-                config.enable_tuned_tensorrt_dynamic_shape(
-                    args.shape_info_filename, True)
-            else:
-                logger.info(
-                    f"when using tensorrt, dynamic shape is a suggested option, you can use '--shape_info_filename=shape.txt' for offline dygnamic shape tuning"
-                )
 
         elif args.use_xpu:
             config.enable_xpu(10 * 1024 * 1024)

tools/infer_kie_token_ser_re.py

@@ -39,7 +39,7 @@ from ppocr.utils.visual import draw_re_results
 from ppocr.utils.logging import get_logger
 from ppocr.utils.utility import get_image_file_list, load_vqa_bio_label_maps, print_dict
 from tools.program import ArgsParser, load_config, merge_config
-from tools.infer_vqa_token_ser import SerPredictor
+from tools.infer_kie_token_ser import SerPredictor
 
 
 class ReArgsParser(ArgsParser):