-
Notifications
You must be signed in to change notification settings - Fork 5.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
13 changed files
with
677 additions
and
23 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,211 @@ | ||
## 使用 C-API 开发预测程序 | ||
|
||
这篇文档通过一个最简单的例子:手写数字识别,来介绍 CPU 下单线程使用 PaddlePaddle C-API 开发预测服务,完整代码见[此目录](https://github.com/PaddlePaddle/Paddle/blob/develop/paddle/capi/examples/model_inference/dense/)。 | ||
|
||
### 使用流程 | ||
|
||
使用 C-API 分为:准备工作和预测程序开发两部分。 | ||
- 准备 | ||
1. 将神经网络模型结构进行序列化。 | ||
- 调用C-API预测时,需要提供序列化之后的网络结构和训练好的模型参数文件。 | ||
1. 将PaddlePaddle训练出的模型参数文件(多个)合并成一个文件。 | ||
- 神经网络模型结构和训练好的模型将被序列化合并入一个文件。 | ||
- 预测时只需加载这一个文件,便于发布。 | ||
- **注意**:以上两种方式只需选择其一即可。 | ||
- 调用 PaddlePaddle C-API 开发预测序 | ||
1. 初始化PaddlePaddle运行环境。 | ||
1. 创建神经网络的输入,组织输入数据。 | ||
1. 加载模型。 | ||
1. 进行前向计算,获得计算结果。 | ||
1. 清理。 | ||
|
||
|
||
这里我们以手写数字识别任务为例,介绍如何使用 C-API 进行预测,完整代码请查看[此目录](https://github.com/PaddlePaddle/Paddle/tree/develop/paddle/capi/examples/model_inference/dense)。 | ||
|
||
运行目录下的 `python mnist_v2.py` 可以使用 PaddlePaddle 内置的 [MNIST 数据集](http://yann.lecun.com/exdb/mnist/)进行训练。脚本中的模型定义了一个简单的含有[两个隐层的全连接网络](https://github.com/PaddlePaddle/book/blob/develop/02.recognize_digits/README.cn.md#softmax回归softmax-regression),网络接受一幅图片作为输入,将图片分类到 0 ~ 9 类别标签之一。训练好的模型默认保存在当前运行目录下的`models`目录中。下面,我们将调用 C-API 加载训练好的模型进行预测。 | ||
|
||
|
||
### 外部准备 | ||
|
||
1. 序列化神经网络模型配置 | ||
|
||
PaddlePaddle 使用 protobuf 来传输网络配置文件中定义的网络结构和相关参数,在使用 C-API 进行预测时,也需将网络结构使用 protobuf 进行序列化,写入文件中。 | ||
|
||
调用`paddle.utils.dump_v2_config`中的`dump_v2_config`函数能够将使用 PaddlePaddle V2 API 定义的神经网络结构 dump 到指定文件中。示例代码如下: | ||
|
||
```python | ||
from paddle.utils.dump_v2_config import dump_v2_config | ||
from mnist_v2 import network | ||
|
||
predict = network(is_infer=True) | ||
dump_v2_config(predict, "trainer_config.bin", True) | ||
``` | ||
|
||
对本例,或运行 `python mnist_v2.py --task dump_config`,会对示例中的网络结构进行序列化,并将结果写入当前目录下的`trainer_config.bin`文件中。 | ||
|
||
当选择使用这种方式调用 C-API 时,如果神经网络有多个可学习参数,请将它们全部放在同一文件夹内,C-API会从指定的目录寻找并加载训练好的模型。 | ||
|
||
2. 合并模型文件(可选) | ||
|
||
一些情况下为了便于发布,希望能够将序列化后的神经网络结构和训练好的模型参数打包进一个文件,这时可以使用`paddle.utils.merge_model`中的`merge_v2_model`接口对神经网络结构和训练好的参数进行序列化,将序列化结果写入一个文件内,调用C-API时直接只需加载这一个文件。 | ||
|
||
代码示例如下: | ||
|
||
```python | ||
from paddle.utils.merge_model import merge_v2_model | ||
|
||
from mnist_v2 import network | ||
|
||
net = network(is_infer=True) | ||
param_file = "models/params_pass_4.tar" | ||
output_file = "output.paddle.model" | ||
merge_v2_model(net, param_file, output_file) | ||
``` | ||
对本例,或者直接运行 `python merge_v2_model.py`,序列化结果将会写入当前目录下的`output.paddle.model`文件中,该文件在调用C-API时,可被直接加载。 | ||
|
||
#### 注意事项 | ||
1. C-API 需要序列化之后神经网络结构,在调用`dump_v2_config`时,参数`binary`必须指定为`True`。 | ||
1. **预测使用的网络结构往往不同于训练**,通常需要去掉网络中的:(1)类别标签层;(2)损失函数层;(3)`evaluator`等,只留下核心计算层,请注意是否需要修改网络结构。 | ||
1. 预测时,可以获取网络中定义的任意多个(大于等于一个)层前向计算的结果,需要哪些层的计算结果作为输出,就将这些层加入一个Python list中,作为调用`dump_v2_config`的第一个参数。 | ||
|
||
### 编写预测代码 | ||
|
||
#### step 1. 初始化及加载模型 | ||
|
||
1. 初始化PaddlePaddle运行环境。 | ||
```c | ||
// Initalize the PaddlePaddle runtime environment. | ||
char* argv[] = {"--use_gpu=False"}; | ||
CHECK(paddle_init(1, (char**)argv)); | ||
``` | ||
|
||
1. 加载训练好的模型。 | ||
|
||
这里需要介绍C-API使用中的一个重要概念:Gradient Machine。概念上,在 PaddlePaddle 内部,一个GradientMachine类的对象管理着一组计算层(PaddlePaddle Layers)来完成前向和反向计算,并处理与之相关的所有细节。特别的,在调用C-API预测时只需进行前向计算。这篇文档的之后部分我们会使用`gradient machine`来特指调用PaddlePaddle C-API创建的GradientMachine类的对象。 | ||
|
||
每一个 `gradient machine` 都会管理维护一份训练好的模型,模型可以通过以下两种方式获取: | ||
1. 从磁盘加载;这时`gradient machine`会独立拥有一份训练好的模型; | ||
1. 共享自其它`gradient machine`的模型;这种情况多出现在使用多线程预测时; | ||
|
||
下面的代码片段创建 `gradient machine`,并从指定路径加载训练好的模型。 | ||
|
||
```c | ||
// Read the binary configuration file generated by `convert_protobin.sh` | ||
long size; | ||
void* buf = read_config(CONFIG_BIN, &size); | ||
|
||
// Create the gradient machine for inference. | ||
paddle_gradient_machine machine; | ||
CHECK(paddle_gradient_machine_create_for_inference(&machine, buf, (int)size)); | ||
|
||
// Load the trained model. Modify the parameter MODEL_PATH to set the correct | ||
// path of the trained model. | ||
CHECK(paddle_gradient_machine_load_parameter_from_disk(machine, MODEL_PATH)); | ||
``` | ||
|
||
##### 注意事项 | ||
1. 以上代码片段使用“仅序列化神经网络结构”的方式加载模型,需要同时指定模型参数存储的路径。 | ||
- 使用PaddlePaddle V2 API训练,模型中所有可学习参数会被存为一个压缩文件,需要手动进行解压,将它们放在同一目录中,C-API不会直接加载 V2 API 存储的压缩文件。 | ||
1. 如果使用`merge model`方式将神经网络结构和训练好的参数序列化到一个文件,请参考此[示例](https://github.com/PaddlePaddle/Mobile/blob/develop/Demo/linux/paddle_image_recognizer.cpp#L59)。 | ||
|
||
#### step 2. 创建神经网络输入,组织输入数据 | ||
|
||
基本使用概念: | ||
- 在PaddlePaddle内部,神经网络中一个计算层的输入/输出被组织为一个 `Argument` 结构体,如果神经网络有多个输入或者多个输入,每一个输入/输入都会对应有自己的`Argument`。 | ||
- `Argument` 并不真正“存储”数据,而是将输入/输出数据有机地组织在一起。 | ||
- 在`Argument`内部由:1. `Matrix`(二维矩阵,存储浮点类型输入/输出);2. `IVector`(一维数组,**仅用于存储整型值**,多用于自然语言处理任务)来实际存储数据。 | ||
|
||
*注:这篇文档使用的示例任务手写数字识别不涉及一维整型序列输入/输出,因此不讨论一维整型输入/输出数据相关的内容。更多信息请参考:[输入数据组织](organization_of_the_inputs.md)。* | ||
|
||
在这篇文档的后面部分,我们会使用`argument`来**特指** PaddlePaddle C-API中神经网的一个输入/输出,使用`matrix`**特指**`argument`中用于存储数据的`Matrix`类的对象,用`ivector`特指`argument`中用于存储数据的`IVector`类的对象。 | ||
|
||
于是,在组织神经网络输入,获取输出时,需要思考完成以下工作: | ||
1. 为每一个输入/输出创建`argument`; | ||
1. 为每一个`argument`创建`matrix`或者`ivector`来存储数据; | ||
|
||
与输入不同的是,输出`argument`的`matrix`变量并不需在使用C-API时为之要分配存储空间。PaddlePaddle内部,神经网络进行前向计算时会自己分配/管理每个计算层的存储空间;这些细节C-API会代为处理,只需在概念上理解,并按照约定调用相关的 C-API 接口即可。 | ||
|
||
下面是示例代码片段。在这段代码中,生成了一条随机输入数据作为测试样本。 | ||
```c | ||
// Inputs and outputs of the network are organized as paddle_arguments object | ||
// in C-API. In the comments below, "argument" specifically means one input of | ||
// the neural network in PaddlePaddle C-API. | ||
paddle_arguments in_args = paddle_arguments_create_none(); | ||
|
||
// There is only one data layer in this demo MNIST network, invoke this | ||
// function to create one argument. | ||
CHECK(paddle_arguments_resize(in_args, 1)); | ||
|
||
// Each argument needs one matrix or one ivector (integer vector, for sparse | ||
// index input, usually used in NLP task) to holds the real input data. | ||
// In the comments below, "matrix" specifically means the object needed by | ||
// argument to hold the data. Here we create the matrix for the above created | ||
// agument to store the testing samples. | ||
paddle_matrix mat = | ||
paddle_matrix_create(/* height = batch size */ 1, | ||
/* width = dimensionality of the data layer */ 784, | ||
/* whether to use GPU */ false); | ||
|
||
paddle_real* array; | ||
// Get the pointer pointing to the start address of the first row of the | ||
// created matrix. | ||
CHECK(paddle_matrix_get_row(mat, 0, &array)); | ||
|
||
// Fill the matrix with a randomly generated test sample. | ||
srand(time(0)); | ||
for (int i = 0; i < 784; ++i) { | ||
array[i] = rand() / ((float)RAND_MAX); | ||
} | ||
|
||
// Assign the matrix to the argument. | ||
CHECK(paddle_arguments_set_value(in_args, 0, mat)); | ||
``` | ||
|
||
#### step 3. 前向计算 | ||
|
||
完成上述准备之后,通过调用 `paddle_gradient_machine_forward` 接口完成神经网络的前向计算。 | ||
示例代码片段如下: | ||
|
||
```c | ||
// Create the output argument. | ||
paddle_arguments out_args = paddle_arguments_create_none(); | ||
|
||
// Invoke the forward computation. | ||
CHECK(paddle_gradient_machine_forward(machine, | ||
in_args, | ||
out_args, | ||
/* is train taks or not */ false)); | ||
|
||
// Create the matrix to hold the forward result of the neural network. | ||
paddle_matrix prob = paddle_matrix_create_none(); | ||
// Access the matrix of the output argument, the predicted result is stored in | ||
// which. | ||
CHECK(paddle_arguments_get_value(out_args, 0, prob)); | ||
|
||
uint64_t height; | ||
uint64_t width; | ||
CHECK(paddle_matrix_get_shape(prob, &height, &width)); | ||
CHECK(paddle_matrix_get_row(prob, 0, &array)); | ||
|
||
printf("Prob: \n"); | ||
for (int i = 0; i < height * width; ++i) { | ||
printf("%.4f ", array[i]); | ||
if ((i + 1) % width == 0) { | ||
printf("\n"); | ||
} | ||
} | ||
printf("\n"); | ||
``` | ||
|
||
#### step 4. 清理 | ||
|
||
结束预测之后,对使用的中间变量和资源进行清理和释放: | ||
|
||
```c | ||
// The cleaning up. | ||
CHECK(paddle_matrix_destroy(prob)); | ||
CHECK(paddle_arguments_destroy(out_args)); | ||
CHECK(paddle_matrix_destroy(mat)); | ||
CHECK(paddle_arguments_destroy(in_args)); | ||
CHECK(paddle_gradient_machine_destroy(machine)); | ||
``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,68 @@ | ||
## 编译 PaddlePaddle 链接库 | ||
|
||
使用 C-API 进行预测依赖于将 PaddlePaddle 核心代码编译成链接库,只需在编译时指定编译选项:`-DWITH_C_API=ON`。同时,**建议将:`DWITH_PYTHON`,`DWITH_SWIG_PY`,`DWITH_GOLANG`,均设置为`OFF`**,以避免链接不必要的库。其它编译选项按需进行设定。 | ||
|
||
```shell | ||
INSTALL_PREFIX=/path/of/capi/ | ||
PADDLE_ROOT=/path/of/paddle_source/ | ||
cmake $PADDLE_ROOT -DCMAKE_INSTALL_PREFIX=$INSTALL_PREFIX \ | ||
-DCMAKE_BUILD_TYPE=Release \ | ||
-DWITH_C_API=ON \ | ||
-DWITH_SWIG_PY=OFF \ | ||
-DWITH_GOLANG=OFF \ | ||
-DWITH_PYTHON=OFF \ | ||
-DWITH_MKLML=OFF \ | ||
-DWITH_MKLDNN=OFF \ | ||
-DWITH_GPU=OFF \ | ||
... | ||
``` | ||
在上面的代码片段中,`PADDLE_ROOT` 表示 PaddlePaddle 源码所在目录,生成Makefile文件后执行:`make && make install`。成功执行后,使用CAPI所需的依赖(包括:(1)编译出的PaddlePaddle 链接和头文件;(2)第三方链接库和头文件)均会存放于`INSTALL_PREFIX`目录中。 | ||
|
||
编译成功后在 `INSTALL_PREFIX` 下会看到如下目录结构(包括了编译出的PaddlePaddle头文件和链接库,以及第三方依赖链接库和头文件(如果需要,由链接方式决定)): | ||
|
||
```text | ||
├── include | ||
│ └── paddle | ||
│ ├── arguments.h | ||
│ ├── capi.h | ||
│ ├── capi_private.h | ||
│ ├── config.h | ||
│ ├── error.h | ||
│ ├── gradient_machine.h | ||
│ ├── main.h | ||
│ ├── matrix.h | ||
│ ├── paddle_capi.map | ||
│ └── vector.h | ||
├── lib | ||
│ ├── libpaddle_capi_engine.a | ||
│ ├── libpaddle_capi_layers.a | ||
│ ├── libpaddle_capi_shared.dylib | ||
│ └── libpaddle_capi_whole.a | ||
└── third_party | ||
├── ...... | ||
``` | ||
|
||
## 链接方式说明 | ||
|
||
目前提供三种链接方式: | ||
|
||
1. 链接`libpaddle_capi_shared.so` 动态库 | ||
- 使用 PaddlePaddle C-API 开发预测程序链接`libpaddle_capi_shared.so`时,需注意: | ||
1. 如果编译时指定编译CPU版本,且使用`OpenBLAS`矩阵库,在使用CAPI开发预测程序时,只需要链接`libpaddle_capi_shared.so`这一个库。 | ||
1. 如果是用编译时指定CPU版本,且使用`MKL`矩阵库,由于`MKL`库有自己独立的动态库文件,在使用PaddlePaddle CAPI开发预测程序时,需要自己链接MKL链接库。 | ||
1. 如果编译时指定编译GPU版本,CUDA相关库会在预测程序运行时动态装载,需要将CUDA相关的库设置到`LD_LIBRARY_PATH`环境变量中。 | ||
- 这种方式最为简便,链接相对容易,**在无特殊需求情况下,推荐使用此方式**。 | ||
|
||
2. 链接静态库 `libpaddle_capi_whole.a` | ||
- 使用PaddlePaddle C-API 开发预测程序链接`libpaddle_capi_whole.a`时,需注意: | ||
1. 需要指定`-Wl,--whole-archive`链接选项。 | ||
1. 需要显式地链接 `gflags`、`glog`、`libz`、`protobuf` 等第三方库,可在`INSTALL_PREFIX\third_party`下找到。 | ||
1. 如果在编译 C-API 时使用OpenBLAS矩阵库,需要显示地链接`libopenblas.a`。 | ||
1. 如果在编译 C-API 是使用 MKL 矩阵库,需要显示地链接 MKL 的动态库。 | ||
|
||
3. 链接静态库 `libpaddle_capi_layers.a`和`libpaddle_capi_engine.a` | ||
- 使用PaddlePaddle C-API 开发预测程序链接`libpaddle_capi_whole.a`时,需注意: | ||
1. 这种链接方式主要用于移动端预测。 | ||
1. 为了减少生成链接库的大小把`libpaddle_capi_whole.a`拆成以上两个静态链接库。 | ||
1. 需指定`-Wl,--whole-archive -lpaddle_capi_layers` 和 `-Wl,--no-whole-archive -lpaddle_capi_engine` 进行链接。 | ||
1. 第三方依赖库需要按照与方式2同样方法显示地进行链接。 |
Empty file.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Oops, something went wrong.