Windows 平台下,我们使用Visual Studio 2019 Community 进行了测试。微软从Visual Studio 2017开始即支持直接管理CMake跨平台编译项目,但是直到2019才提供了稳定和完全的支持,所以如果你想使用CMake管理项目编译构建,我们推荐你使用Visual Studio 2019环境下构建。
请确保系统已经安装好上述基本软件,我们使用的是VS2019的社区版。
下面所有示例以工作目录为 D:\projects演示。
d:
mkdir projects
cd projects
git clone https://github.com/PaddlePaddle/PaddleX.git
说明:其中C++预测代码在PaddleX\deploy\cpp 目录,该目录不依赖任何PaddleX下其他目录。
PaddlePaddle C++ 预测库针对是否使用GPU、是否支持TensorRT、以及不同的CUDA版本提供了已经编译好的预测库,目前PaddleX依赖于Paddle 1.8,基于Paddle 1.8的Paddle预测库下载链接如下所示:
| 版本说明 | 预测库(1.8.2版本) | 编译器 | 构建工具 | cuDNN | CUDA |
|---|---|---|---|---|---|
| cpu_avx_mkl | paddle_inference | MSVC 2015 update 3 | CMake v3.16.0 | ||
| cpu_avx_openblas | paddle_inference | MSVC 2015 update 3 | CMake v3.16.0 | ||
| cuda9.0_cudnn7_avx_mkl | paddle_inference | MSVC 2015 update 3 | CMake v3.16.0 | 7.4.1 | 9.0 |
| cuda9.0_cudnn7_avx_openblas | paddle_inference | MSVC 2015 update 3 | CMake v3.16.0 | 7.4.1 | 9.0 |
| cuda10.0_cudnn7_avx_mkl | paddle_inference | MSVC 2015 update 3 | CMake v3.16.0 | 7.5.0 | 9.0 |
请根据实际情况选择下载,如若以上版本不满足您的需求,请至C++预测库下载列表选择符合的版本。
将预测库解压后,其所在目录(例如D:\projects\fluid_inference\)下主要包含的内容有:
├── \paddle\ # paddle核心库和头文件
|
├── \third_party\ # 第三方依赖库和头文件
|
└── \version.txt # 版本和编译信息
D:\projects\opencvD:\projects\opencv\build\x64\vc14\bin继续但无需代码文件->打开->CMake选择C++预测代码所在路径(例如D:\projects\PaddleX\deploy\cpp),并打开CMakeList.txt:
项目->CMake设置浏览,分别设置编译选项指定CUDA、OpenCV、Paddle预测库的路径依赖库路径的含义说明如下(带*表示仅在使用GPU版本预测库时指定, 其中CUDA库版本尽量与Paddle预测库的对齐,例如Paddle预测库是使用9.0、10.0版本编译的,则编译PaddleX预测代码时不使用9.2、10.1等版本CUDA库):
| 参数名 | 含义 |
|---|---|
| *CUDA_LIB | CUDA的库路径, 注:请将CUDNN的cudnn.lib文件拷贝到CUDA_LIB路径下 |
| OPENCV_DIR | OpenCV的安装路径, |
| PADDLE_DIR | Paddle c++预测库的路径 |
注意:
CPU版预测库,请把WITH_GPU的值去掉勾openblas版本,请把WITH_MKL的值去掉勾cmake/yaml.cmake中将URL https://bj.bcebos.com/paddlex/deploy/deps/yaml-cpp.zip 中的网址,改为下载文件的路径。D:/projects,解压后目录为D:/projects/paddlex-encryption。编译时需勾选WITH_EBNCRYPTION并且在ENCRTYPTION_DIR填入D:/projects/paddlex-encryption。
设置完成后, 点击上图中保存并生成CMake缓存以加载变量。生成->全部生成在加载模型前,请检查你的模型目录中文件应该包括model.yml、__model__和__params__三个文件。如若不满足这个条件,请参考部署模型导出将模型导出为部署格式。
上述Visual Studio 2019编译产出的可执行文件在out\build\x64-Release目录下,打开cmd,并切换到该目录:
D:
cd D:\projects\PaddleX\deploy\cpp\out\build\x64-Release
编译成功后,预测demo的入口程序为paddlex_inference\detector.exe,paddlex_inference\classifier.exe,paddlex_inference\segmenter.exe,用户可根据自己的模型类型选择,其主要命令参数说明如下:
| 参数 | 说明 |
|---|---|
| model_dir | 导出的预测模型所在路径 |
| image | 要预测的图片文件路径 |
| image_list | 按行存储图片路径的.txt文件 |
| use_gpu | 是否使用 GPU 预测, 支持值为0或1(默认值为0) |
| gpu_id | GPU 设备ID, 默认值为0 |
| save_dir | 保存可视化结果的路径, 默认值为"output",classifier无该参数 |
| key | 加密过程中产生的密钥信息,默认值为""表示加载的是未加密的模型 |
| batch_size | 预测的批量大小,默认为1 |
| thread_num | 预测的线程数,默认为cpu处理器个数 |
| use_ir_optim | 是否使用图优化策略,支持值为0或1(默认值为1,图像分割默认值为0) |
可使用小度熊识别模型中导出的inference_model和测试图片进行预测, 例如导出到D:\projects,模型路径为D:\projects\inference_model。
关于预测速度的说明:加载模型后前几张图片的预测速度会较慢,这是因为运行启动时涉及到内存显存初始化等步骤,通常在预测20-30张图片后模型的预测速度达到稳定。
不使用GPU测试图片 D:\images\xiaoduxiong.jpeg
.\paddlex_inference\detector.exe --model_dir=D:\projects\inference_model --image=D:\images\xiaoduxiong.jpeg --save_dir=output
图片文件可视化预测结果会保存在save_dir参数设置的目录下。
使用GPU预测多个图片D:\images\image_list.txt,image_list.txt内容的格式如下:
D:\images\xiaoduxiong1.jpeg
D:\images\xiaoduxiong2.jpeg
...
D:\images\xiaoduxiongn.jpeg
.\paddlex_inference\detector.exe --model_dir=D:\projects\inference_model --image_list=D:\images\image_list.txt --use_gpu=1 --save_dir=output --batch_size=2 --thread_num=2
图片文件可视化预测结果会保存在save_dir参数设置的目录下。
如果未对模型进行加密,请参考加密PaddleX模型对模型进行加密。例如加密后的模型所在目录为D:\projects\encrypted_inference_model。
.\paddlex_inference\detector.exe --model_dir=D:\projects\encrypted_inference_model --image=D:\images\xiaoduxiong.jpeg --save_dir=output --key=kLAl1qOs5uRbFt0/RrIDTZW2+tOf5bzvUIaHGF8lJ1c=
--key传入加密工具输出的密钥,例如kLAl1qOs5uRbFt0/RrIDTZW2+tOf5bzvUIaHGF8lJ1c=, 图片文件可视化预测结果会保存在save_dir参数设置的目录下。