소스 검색

update paddlex_cli

cuicheng01 10 달 전
부모
커밋
ad43de32eb
1개의 변경된 파일8개의 추가작업 그리고 6개의 파일을 삭제
  1. 8 6
      paddlex/paddlex_cli.py

+ 8 - 6
paddlex/paddlex_cli.py

@@ -280,12 +280,14 @@ def main():
             interactive_get_pipeline(args.get_pipeline_config, args.save_path)
         else:
             pipeline_args_dict = {}
-            for arg in pipeline_args:
-                arg_name = arg["name"].lstrip("-")
-                if hasattr(args, arg_name):
-                    pipeline_args_dict[arg_name] = getattr(args, arg_name)
-                else:
-                    logging.warning(f"Argument {arg_name} is missing in args")
+            from .utils.flags import USE_NEW_INFERENCE
+            if USE_NEW_INFERENCE:
+                for arg in pipeline_args:
+                    arg_name = arg["name"].lstrip("-")
+                    if hasattr(args, arg_name):
+                        pipeline_args_dict[arg_name] = getattr(args, arg_name)
+                    else:
+                        logging.warning(f"Argument {arg_name} is missing in args")
             return pipeline_predict(
                 args.pipeline,
                 args.input,