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++ 预测库针对不同的CPU,CUDA,以及是否支持TensorRT,提供了不同的预编译版本,目前PaddleX依赖于Paddle1.7版本,以下提供了多个不同版本的Paddle预测库:
| 版本说明 | 预测库(1.7.2版本) | 编译器 | 构建工具 | cuDNN | CUDA |
|---|---|---|---|---|---|
| cpu_avx_mkl | fluid_inference.zip | MSVC 2015 update 3 | CMake v3.16.0 | ||
| cpu_avx_openblas | fluid_inference.zip | MSVC 2015 update 3 | CMake v3.16.0 | ||
| cuda9.0_cudnn7_avx_mkl | fluid_inference.zip | MSVC 2015 update 3 | CMake v3.16.0 | 7.4.1 | 9.0 |
| cuda9.0_cudnn7_avx_openblas | fluid_inference.zip | MSVC 2015 update 3 | CMake v3.16.0 | 7.4.1 | 9.0 |
| cuda10.0_cudnn7_avx_mkl | fluid_inference.zip | 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选择项目代码所在路径,并打开CMakeList.txt:
项目->PADDLEX_INFERENCE的CMake设置浏览,分别设置编译选项指定CUDA、OpenCV、Paddle预测库的路径依赖库路径的含义说明如下(带*表示仅在使用GPU版本预测库时指定, 其中CUDA库版本尽量对齐,使用9.0、10.0版本,不使用9.2、10.1等版本CUDA库):
| 参数名 | 含义 |
|---|---|
| *CUDA_LIB | CUDA的库路径, 注:请将CUDNN的cudnn.lib文件拷贝到CUDA_LIB路径下 |
| OPENCV_DIR | OpenCV的安装路径, |
| PADDLE_DIR | Paddle c++预测库的路径 |
注意: 1. 使用CPU版预测库,请把WITH_GPU的值去掉勾 2. 如果使用的是openblas版本,请把WITH_MKL的值去掉勾
设置完成后, 点击上图中保存并生成CMake缓存以加载变量。
生成->全部生成上述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\classifer.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",classfier无该参数 |
可使用小度熊识别模型中导出的inference_model和测试图片进行预测。
样例一:
不使用GPU测试图片 \\path\\to\\xiaoduxiong.jpeg
.\\paddlex_inference\\detector.exe --model_dir=\\path\\to\\inference_model --image=D:\\images\\xiaoduxiong.jpeg --save_dir=output
图片文件可视化预测结果会保存在save_dir参数设置的目录下。
样例二:
使用GPU预测多个图片\\path\\to\\image_list.txt,image_list.txt内容的格式如下:
\\path\\to\\images\\xiaoduxiong1.jpeg
\\path\\to\\images\\xiaoduxiong2.jpeg
...
\\path\\to\\images\\xiaoduxiongn.jpeg
.\\paddlex_inference\\detector.exe --model_dir=\\path\\to\\inference_model --image_list=\\path\\to\\images_list.txt --use_gpu=1 --save_dir=output
图片文件可视化预测结果会保存在save_dir参数设置的目录下。