README.md 2.2 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 16 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_server==0.3.2 #CPU
pip install paddle_serving_server_gpu==0.3.2.post9 #GPU with CUDA9.0
pip install paddle_serving_server_gpu==0.3.2.post9 #GPU with CUDA10.0
```

客户端安装:

```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
## 4.部署预测服务

```shell
M
MRXLT 已提交
55 56
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 已提交
57 58 59
```

## 5.执行预测
M
MRXLT 已提交
60 61
```shell
python seg_client.py ../../dataset/optic_disc_seg/JPEGImages/N0060.jpg
M
MRXLT 已提交
62
```
M
MRXLT 已提交
63 64
脚本执行之后,会在输入图片所在的目录下生成处理后的图片
示例中为`../../dataset/optic_disc_seg/JPEGImages/N0060_jpg_mask.png``../../dataset/optic_disc_seg/JPEGImages/N0060_jpg_result.png`