README.md 2.8 KB
Newer Older
M
MRXLT 已提交
1 2 3 4 5 6 7 8
# 通过PaddleServing部署服务

## 1.简介

PaddleServing是Paddle的在线预测服务框架,可以快速部署训练好的模型用于在线预测。更多信息请参考[PaddleServing 主页](https://github.com/PaddlePaddle/Serving)。本文中将通过unet模型示例,展示预测服务的部署和预测过程。

## 2.安装Paddle Serving

M
MRXLT 已提交
9
目前PaddleServing的正式版本为0.3.2版本。
M
MRXLT 已提交
10 11 12 13 14 15

服务端安装:

```shell
pip install paddle_serving_server==0.3.2 #CPU
pip install paddle_serving_server_gpu==0.3.2.post9 #GPU with CUDA9.0
M
MRXLT 已提交
16
pip install paddle_serving_server_gpu==0.3.2.post10 #GPU with CUDA10.0
M
MRXLT 已提交
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
```

客户端安装:

```shell
pip install paddle_serving_client==0.3.2
```

## 3.导出预测模型


通过训练得到一个满足要求的模型后,如果想要将该模型接入到PaddleServing服务,我们需要通过[`pdseg/export_serving_model.py`](../../pdseg/export_serving_model.py)来导出该模型。

该脚本的使用方法和`train.py/eval.py/vis.py`完全一样。

### FLAGS

|FLAG|用途|默认值|备注|
|-|-|-|-|
|--cfg|配置文件路径|None||

### 使用示例

我们使用[训练/评估/可视化](./usage.md)一节中训练得到的模型进行试用,命令如下

```shell
python pdseg/export_serving_model.py --cfg configs/unet_optic.yaml TEST.TEST_MODEL ./saved_model/unet_optic/final
```

预测模型会导出到`freeze_model`目录,包括`serving_server``serving_client`两个子目录。

`freeze_model/serving_server`目录下包含了模型文件和serving server端配置文件,`freeze_model/serving_client`目录下包含了serving client端配置文件。

M
MRXLT 已提交
50 51
分别将serving_server和serving_client复制到server和client启动的路径下。

M
MRXLT 已提交
52 53 54 55
本文中导出的unet模型示例[下载](https://paddle-serving.bj.bcebos.com/paddle_seg_demo/seg_unet_demo.tar.gz)

解压下载后的压缩包,可以得到seving_server和serving_client两个文件夹,用于以下步骤的测试。

M
MRXLT 已提交
56 57 58
## 4.部署预测服务

```shell
M
MRXLT 已提交
59 60
python -m paddle_serving_server.serve --model serving_server/ --port 9494 # CPU
python -m paddle_serving_server_gpu.serve --model serving_server --port 9494 --gpu_ids 0 #GPU
M
MRXLT 已提交
61 62 63
```

## 5.执行预测
M
MRXLT 已提交
64 65
```shell
python seg_client.py ../../dataset/optic_disc_seg/JPEGImages/N0060.jpg
M
MRXLT 已提交
66
```
M
MRXLT 已提交
67 68
脚本执行之后,会在输入图片所在的目录下生成处理后的图片
示例中为`../../dataset/optic_disc_seg/JPEGImages/N0060_jpg_mask.png``../../dataset/optic_disc_seg/JPEGImages/N0060_jpg_result.png`
M
MRXLT 已提交
69 70 71 72 73 74

如果需要使用其他模型进行预测,需要修改预处理部分和后处理部分。

本文中使用的是paddle_serving_app中内置的处理方法,用户可以参考导出模型时生成的deploy.yaml修改预处理部分。

后处理部分SegPostprocess初始化时接收的参数为分割的类别数。