未验证 提交 56128875 编写于 作者: J Jason 提交者: GitHub

Merge pull request #72 from PaddlePaddle/devleop_deploy

add deploy python code
# paddlex.deploy # Predictor部署-paddlex.deploy
使用AnalysisPredictor进行预测部署。 使用AnalysisPredictor进行预测部署。
## create_predictor ## Predictor类
``` ```
paddlex.deploy.create_predictor(model_dir, use_gpu=False) paddlex.deploy.Predictor(model_dir, use_gpu=False, gpu_id=0, use_mkl=False, use_trt=False, use_glog=False, memory_optimize=True)
``` ```
#### Args > **参数**
* **model_dir**: 训练过程中保存的模型路径 > > * **model_dir**: 训练过程中保存的模型路径, 注意需要使用导出的inference模型
* **use_gpu**: 是否使用GPU进行预测 > > * **use_gpu**: 是否使用GPU进行预测
> > * **gpu_id**: 使用的GPU序列号
> > * **use_mkl**: 是否使用mkldnn加速库
> > * **use_trt**: 是否使用TensorRT预测引擎
> > * **use_glog**: 是否打印中间日志
> > * **memory_optimize**: 是否优化内存使用
#### Returns > > ### 示例
> >
> > ```
> > import paddlex
> > model = paddlex.deploy.Predictor(model_dir, use_gpu=True)
> > result = model.predict(image_file)
> > ```
* **Predictor**: paddlex.deploy.predictor.Predictor ### predict 接口
> ```
> predict(image, topk=1)
> ```
### 示例 > **参数
``` * **image(str|np.ndarray)**: 待预测的图片路径或np.ndarray,若为后者需注意为BGR格式
import paddlex * **topk(int)**: 图像分类时使用的参数,表示预测前topk个可能的分类
# 下
Predictor = paddlex.deploy.create_predictor(model_dir, use_gpu=True)
```
## ClassifyPredictor
继承至paddlex.deploy.predictor.Predictor,当model_dir/model.yml里面
```
paddlex.deploy.create_predictor(model_dir, use_gpu=False)
```
#### Args
* **model_dir**: 训练过程中保存的模型路径
* **use_gpu**: 是否使用GPU进行预测
#### Returns
* **Predictor**: paddlex.deploy.predictor.Predictor
### 示例
```
import paddlex
# 下
Predictor = paddlex.deploy.create_predictor(model_dir, use_gpu=True)
```
...@@ -10,3 +10,4 @@ PaddleX API说明文档 ...@@ -10,3 +10,4 @@ PaddleX API说明文档
slim.md slim.md
load_model.md load_model.md
visualize.md visualize.md
deploy.md
多端部署 多端部署
============== ==============
.. toctree::
:maxdepth: 2
:caption: 文档目录:
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册