Prechádzať zdrojové kódy

use default official model in predict mode when specify model_dir is None

gaotingquan 1 rok pred
rodič
commit
aa5ade4bfe

+ 1 - 1
paddlex/configs/table_recognition/SLANet.yaml

@@ -33,7 +33,7 @@ Export:
 
 Predict:
   model_dir: "output/best_accuracy"
-  input_path: "/paddle/dataset/paddlex/table_rec/table_rec_dataset_examples/images/border_10368_GBUAFQNHRKR5FUQ6ZE50.jpg"
+  input_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/table_recognition.jpg"
   kernel_option:
     run_mode: paddle
     batch_size: 1

+ 3 - 0
paddlex/modules/base/predictor/predictor.py

@@ -171,6 +171,9 @@ class PredictorBuilderByConfig(object):
 
         predict_config = deepcopy(config.Predict)
         model_dir = predict_config.pop("model_dir")
+        if model_dir is None:
+            if model_name in official_models:
+                model_dir = official_models[model_name]
         kernel_setting = predict_config.pop("kernel_option", {})
         kernel_setting.setdefault("device", device)
         kernel_option = PaddleInferenceOption(**kernel_setting)