소스 검색

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__)