ソースを参照

support save in model predict

gaotingquan 1 年間 前
コミット
ce6205608d
1 ファイル変更2 行追加0 行削除
  1. 2 0
      paddlex/engine.py

+ 2 - 0
paddlex/engine.py

@@ -45,5 +45,7 @@ class Engine(object):
         elif self._mode == "predict":
             for res in self._model.predict():
                 res.print(json_format=False)
+            if self._output:
+                res.save_all(save_path=self._output)
         else:
             raise_unsupported_api_error(f"{self._mode}", self.__class__)