更换文档检测模型
This commit is contained in:
210
paddle_detection/deploy/cpp/docs/Jetson_build.md
Normal file
210
paddle_detection/deploy/cpp/docs/Jetson_build.md
Normal file
@@ -0,0 +1,210 @@
|
||||
# Jetson平台编译指南
|
||||
|
||||
## 说明
|
||||
`NVIDIA Jetson`设备是具有`NVIDIA GPU`的嵌入式设备,可以将目标检测算法部署到该设备上。本文档是在`Jetson`硬件上部署`PaddleDetection`模型的教程。
|
||||
|
||||
本文档以`Jetson TX2`硬件、`JetPack 4.3`版本为例进行说明。
|
||||
|
||||
`Jetson`平台的开发指南请参考[NVIDIA Jetson Linux Developer Guide](https://docs.nvidia.com/jetson/l4t/index.html).
|
||||
|
||||
## Jetson环境搭建
|
||||
`Jetson`系统软件安装,请参考[NVIDIA Jetson Linux Developer Guide](https://docs.nvidia.com/jetson/l4t/index.html).
|
||||
|
||||
* (1) 查看硬件系统的l4t的版本号
|
||||
```
|
||||
cat /etc/nv_tegra_release
|
||||
```
|
||||
* (2) 根据硬件,选择硬件可安装的`JetPack`版本,硬件和`JetPack`版本对应关系请参考[jetpack-archive](https://developer.nvidia.com/embedded/jetpack-archive).
|
||||
|
||||
* (3) 下载`JetPack`,请参考[NVIDIA Jetson Linux Developer Guide](https://docs.nvidia.com/jetson/l4t/index.html) 中的`Preparing a Jetson Developer Kit for Use`章节内容进行刷写系统镜像。
|
||||
|
||||
**注意**: 请在[jetpack-archive](https://developer.nvidia.com/embedded/jetpack-archive) 根据硬件选择适配的`JetPack`版本进行刷机。
|
||||
|
||||
## 下载或编译`Paddle`预测库
|
||||
本文档使用`Paddle`在`JetPack4.3`上预先编译好的预测库,请根据硬件在[安装与编译 Linux 预测库](https://www.paddlepaddle.org.cn/documentation/docs/zh/guides/05_inference_deployment/inference/build_and_install_lib_cn.html) 中选择对应版本的`Paddle`预测库。
|
||||
|
||||
这里选择[nv_jetson_cuda10_cudnn7.6_trt6(jetpack4.3)](https://paddle-inference-lib.bj.bcebos.com/2.0.0-nv-jetson-jetpack4.3-all/paddle_inference.tgz), `Paddle`版本`2.0.0-rc0`,`CUDA`版本`10.0`,`CUDNN`版本`7.6`,`TensorRT`版本`6`。
|
||||
|
||||
若需要自己在`Jetson`平台上自定义编译`Paddle`库,请参考文档[安装与编译 Linux 预测库](https://www.paddlepaddle.org.cn/documentation/docs/zh/advanced_guide/inference_deployment/inference/build_and_install_lib_cn.html) 的`NVIDIA Jetson嵌入式硬件预测库源码编译`部分内容。
|
||||
|
||||
### Step1: 下载代码
|
||||
|
||||
`git clone https://github.com/PaddlePaddle/PaddleDetection.git`
|
||||
|
||||
**说明**:其中`C++`预测代码在`/root/projects/PaddleDetection/deploy/cpp` 目录,该目录不依赖任何`PaddleDetection`下其他目录。
|
||||
|
||||
|
||||
### Step2: 下载PaddlePaddle C++ 预测库 paddle_inference
|
||||
|
||||
解压下载的[nv_jetson_cuda10_cudnn7.6_trt6(jetpack4.3)](https://paddle-inference-lib.bj.bcebos.com/2.0.1-nv-jetson-jetpack4.3-all/paddle_inference.tgz) 。
|
||||
|
||||
下载并解压后`/root/projects/paddle_inference`目录包含内容为:
|
||||
```
|
||||
paddle_inference
|
||||
├── paddle # paddle核心库和头文件
|
||||
|
|
||||
├── third_party # 第三方依赖库和头文件
|
||||
|
|
||||
└── version.txt # 版本和编译信息
|
||||
```
|
||||
|
||||
**注意:** 预编译库`nv-jetson-cuda10-cudnn7.6-trt6`使用的`GCC`版本是`7.5.0`,其他都是使用`GCC 4.8.5`编译的。使用高版本的GCC可能存在`ABI`兼容性问题,建议降级或[自行编译预测库](https://www.paddlepaddle.org.cn/documentation/docs/zh/advanced_guide/inference_deployment/inference/build_and_install_lib_cn.html)。
|
||||
|
||||
|
||||
### Step4: 编译
|
||||
|
||||
编译`cmake`的命令在`scripts/build.sh`中,请根据实际情况修改主要参数,其主要内容说明如下:
|
||||
|
||||
注意,`TX2`平台的`CUDA`、`CUDNN`需要通过`JetPack`安装。
|
||||
|
||||
```
|
||||
# 是否使用GPU(即是否使用 CUDA)
|
||||
WITH_GPU=ON
|
||||
|
||||
# 是否使用MKL or openblas,TX2需要设置为OFF
|
||||
WITH_MKL=OFF
|
||||
|
||||
# 是否集成 TensorRT(仅WITH_GPU=ON 有效)
|
||||
WITH_TENSORRT=ON
|
||||
|
||||
# TensorRT 的include路径
|
||||
TENSORRT_INC_DIR=/usr/include/aarch64-linux-gnu
|
||||
|
||||
# TensorRT 的lib路径
|
||||
TENSORRT_LIB_DIR=/usr/lib/aarch64-linux-gnu
|
||||
|
||||
# Paddle 预测库路径
|
||||
PADDLE_DIR=/path/to/paddle_inference/
|
||||
|
||||
# Paddle 预测库名称
|
||||
PADDLE_LIB_NAME=paddle_inference
|
||||
|
||||
# Paddle 的预测库是否使用静态库来编译
|
||||
# 使用TensorRT时,Paddle的预测库通常为动态库
|
||||
WITH_STATIC_LIB=OFF
|
||||
|
||||
# CUDA 的 lib 路径
|
||||
CUDA_LIB=/usr/local/cuda-10.0/lib64
|
||||
|
||||
# CUDNN 的 lib 路径
|
||||
CUDNN_LIB=/usr/lib/aarch64-linux-gnu
|
||||
|
||||
# 是否开启关键点模型预测功能
|
||||
WITH_KEYPOINT=ON
|
||||
|
||||
# OPENCV_DIR 的路径
|
||||
# linux平台请下载:https://bj.bcebos.com/paddleseg/deploy/opencv3.4.6gcc4.8ffmpeg.tar.gz2,并解压到deps文件夹下
|
||||
# TX2平台请下载:https://paddlemodels.bj.bcebos.com/TX2_JetPack4.3_opencv_3.4.10_gcc7.5.0.zip,并解压到deps文件夹下
|
||||
OPENCV_DIR=/path/to/opencv
|
||||
|
||||
# 请检查以上各个路径是否正确
|
||||
|
||||
# 以下无需改动
|
||||
cmake .. \
|
||||
-DWITH_GPU=${WITH_GPU} \
|
||||
-DWITH_MKL=OFF \
|
||||
-DWITH_TENSORRT=${WITH_TENSORRT} \
|
||||
-DTENSORRT_DIR=${TENSORRT_DIR} \
|
||||
-DPADDLE_DIR=${PADDLE_DIR} \
|
||||
-DWITH_STATIC_LIB=${WITH_STATIC_LIB} \
|
||||
-DCUDA_LIB=${CUDA_LIB} \
|
||||
-DCUDNN_LIB=${CUDNN_LIB} \
|
||||
-DOPENCV_DIR=${OPENCV_DIR} \
|
||||
-DPADDLE_LIB_NAME={PADDLE_LIB_NAME} \
|
||||
-DWITH_KEYPOINT=${WITH_KEYPOINT}
|
||||
make
|
||||
```
|
||||
|
||||
例如设置如下:
|
||||
```
|
||||
# 是否使用GPU(即是否使用 CUDA)
|
||||
WITH_GPU=ON
|
||||
|
||||
# 是否使用MKL or openblas
|
||||
WITH_MKL=OFF
|
||||
|
||||
# 是否集成 TensorRT(仅WITH_GPU=ON 有效)
|
||||
WITH_TENSORRT=OFF
|
||||
|
||||
# TensorRT 的include路径
|
||||
TENSORRT_INC_DIR=/usr/include/aarch64-linux-gnu
|
||||
|
||||
# TensorRT 的lib路径
|
||||
TENSORRT_LIB_DIR=/usr/lib/aarch64-linux-gnu
|
||||
|
||||
# Paddle 预测库路径
|
||||
PADDLE_DIR=/home/nvidia/PaddleDetection_infer/paddle_inference/
|
||||
|
||||
# Paddle 预测库名称
|
||||
PADDLE_LIB_NAME=paddle_inference
|
||||
|
||||
# Paddle 的预测库是否使用静态库来编译
|
||||
# 使用TensorRT时,Paddle的预测库通常为动态库
|
||||
WITH_STATIC_LIB=OFF
|
||||
|
||||
# CUDA 的 lib 路径
|
||||
CUDA_LIB=/usr/local/cuda-10.0/lib64
|
||||
|
||||
# CUDNN 的 lib 路径
|
||||
CUDNN_LIB=/usr/lib/aarch64-linux-gnu/
|
||||
|
||||
# 是否开启关键点模型预测功能
|
||||
WITH_KEYPOINT=ON
|
||||
```
|
||||
|
||||
修改脚本设置好主要参数后,执行`build`脚本:
|
||||
```shell
|
||||
sh ./scripts/build.sh
|
||||
```
|
||||
|
||||
### Step5: 预测及可视化
|
||||
编译成功后,预测入口程序为`build/main`其主要命令参数说明如下:
|
||||
| 参数 | 说明 |
|
||||
| ---- | ---- |
|
||||
| --model_dir | 导出的检测预测模型所在路径 |
|
||||
| --model_dir_keypoint | Option | 导出的关键点预测模型所在路径 |
|
||||
| --image_file | 要预测的图片文件路径 |
|
||||
| --image_dir | 要预测的图片文件夹路径 |
|
||||
| --video_file | 要预测的视频文件路径 |
|
||||
| --camera_id | Option | 用来预测的摄像头ID,默认为-1(表示不使用摄像头预测)|
|
||||
| --device | 运行时的设备,可选择`CPU/GPU/XPU`,默认为`CPU`|
|
||||
| --gpu_id | 指定进行推理的GPU device id(默认值为0)|
|
||||
| --run_mode | 使用GPU时,默认为paddle, 可选(paddle/trt_fp32/trt_fp16/trt_int8)|
|
||||
| --batch_size | 检测模型预测时的batch size,在指定`image_dir`时有效 |
|
||||
| --batch_size_keypoint | 关键点模型预测时的batch size,默认为8 |
|
||||
| --run_benchmark | 是否重复预测来进行benchmark测速 |
|
||||
| --output_dir | 输出图片所在的文件夹, 默认为output |
|
||||
| --use_mkldnn | CPU预测中是否开启MKLDNN加速 |
|
||||
| --cpu_threads | 设置cpu线程数,默认为1 |
|
||||
| --use_dark | 关键点模型输出预测是否使用DarkPose后处理,默认为true |
|
||||
|
||||
**注意**:
|
||||
- 优先级顺序:`camera_id` > `video_file` > `image_dir` > `image_file`。
|
||||
- --run_benchmark如果设置为True,则需要安装依赖`pip install pynvml psutil GPUtil`。
|
||||
|
||||
|
||||
`样例一`:
|
||||
```shell
|
||||
#不使用`GPU`测试图片 `/root/projects/images/test.jpeg`
|
||||
./main --model_dir=/root/projects/models/yolov3_darknet --image_file=/root/projects/images/test.jpeg
|
||||
```
|
||||
|
||||
图片文件`可视化预测结果`会保存在当前目录下`output.jpg`文件中。
|
||||
|
||||
|
||||
`样例二`:
|
||||
```shell
|
||||
#使用 `GPU`预测视频`/root/projects/videos/test.mp4`
|
||||
./main --model_dir=/root/projects/models/yolov3_darknet --video_path=/root/projects/images/test.mp4 --device=GPU
|
||||
```
|
||||
视频文件目前支持`.mp4`格式的预测,`可视化预测结果`会保存在当前目录下`output.mp4`文件中。
|
||||
|
||||
`样例三`:
|
||||
```shell
|
||||
#使用关键点模型与检测模型联合预测,使用 `GPU`预测
|
||||
#检测模型检测到的人送入关键点模型进行关键点预测
|
||||
./main --model_dir=/root/projects/models/yolov3_darknet --model_dir_keypoint=/root/projects/models/hrnet_w32_256x192 --image_file=/root/projects/images/test.jpeg --device=GPU
|
||||
```
|
||||
|
||||
## 性能测试
|
||||
benchmark请查看[BENCHMARK_INFER](../../BENCHMARK_INFER.md)
|
||||
149
paddle_detection/deploy/cpp/docs/linux_build.md
Normal file
149
paddle_detection/deploy/cpp/docs/linux_build.md
Normal file
@@ -0,0 +1,149 @@
|
||||
# Linux平台编译指南
|
||||
|
||||
## 说明
|
||||
本文档在 `Linux`平台使用`GCC 8.2`测试过,如果需要使用其他G++版本编译使用,则需要重新编译Paddle预测库,请参考: [从源码编译Paddle预测库](https://paddleinference.paddlepaddle.org.cn/user_guides/source_compile.html)。本文档使用的预置的opencv库是在ubuntu 16.04上用gcc8.2编译的,如果需要在gcc8.2以外的环境编译,那么需自行编译opencv库。
|
||||
|
||||
## 前置条件
|
||||
* G++ 8.2
|
||||
* CUDA 9.0 / CUDA 10.1, cudnn 7+ (仅在使用GPU版本的预测库时需要)
|
||||
* CMake 3.0+
|
||||
|
||||
请确保系统已经安装好上述基本软件,**下面所有示例以工作目录为 `/root/projects/`演示**。
|
||||
|
||||
### Step1: 下载代码
|
||||
|
||||
`git clone https://github.com/PaddlePaddle/PaddleDetection.git`
|
||||
|
||||
**说明**:其中`C++`预测代码在`/root/projects/PaddleDetection/deploy/cpp` 目录,该目录不依赖任何`PaddleDetection`下其他目录。
|
||||
|
||||
|
||||
### Step2: 下载PaddlePaddle C++ 预测库 paddle_inference
|
||||
|
||||
PaddlePaddle C++ 预测库针对不同的`CPU`和`CUDA`版本提供了不同的预编译版本,请根据实际情况下载: [C++预测库下载列表](https://paddleinference.paddlepaddle.org.cn/user_guides/download_lib.html)
|
||||
|
||||
|
||||
下载并解压后`/root/projects/paddle_inference`目录包含内容为:
|
||||
```
|
||||
paddle_inference
|
||||
├── paddle # paddle核心库和头文件
|
||||
|
|
||||
├── third_party # 第三方依赖库和头文件
|
||||
|
|
||||
└── version.txt # 版本和编译信息
|
||||
```
|
||||
|
||||
**注意:** 预编译版本除`nv-jetson-cuda10-cudnn7.5-trt5` 以外其它包都是基于`GCC 4.8.5`编译,使用高版本`GCC`可能存在 `ABI`兼容性问题,建议降级或[自行编译预测库](https://www.paddlepaddle.org.cn/documentation/docs/zh/advanced_guide/inference_deployment/inference/build_and_install_lib_cn.html)。
|
||||
|
||||
|
||||
### Step3: 编译
|
||||
|
||||
编译`cmake`的命令在`scripts/build.sh`中,请根据实际情况修改主要参数,其主要内容说明如下:
|
||||
|
||||
```
|
||||
# 是否使用GPU(即是否使用 CUDA)
|
||||
WITH_GPU=OFF
|
||||
|
||||
# 使用MKL or openblas
|
||||
WITH_MKL=ON
|
||||
|
||||
# 是否集成 TensorRT(仅WITH_GPU=ON 有效)
|
||||
WITH_TENSORRT=OFF
|
||||
|
||||
# TensorRT 的include路径
|
||||
TENSORRT_LIB_DIR=/path/to/TensorRT/include
|
||||
|
||||
# TensorRT 的lib路径
|
||||
TENSORRT_LIB_DIR=/path/to/TensorRT/lib
|
||||
|
||||
# Paddle 预测库路径
|
||||
PADDLE_DIR=/path/to/paddle_inference
|
||||
|
||||
# Paddle 预测库名称
|
||||
PADDLE_LIB_NAME=paddle_inference
|
||||
|
||||
# CUDA 的 lib 路径
|
||||
CUDA_LIB=/path/to/cuda/lib
|
||||
|
||||
# CUDNN 的 lib 路径
|
||||
CUDNN_LIB=/path/to/cudnn/lib
|
||||
|
||||
# 是否开启关键点模型预测功能
|
||||
WITH_KEYPOINT=ON
|
||||
|
||||
# 请检查以上各个路径是否正确
|
||||
|
||||
# 以下无需改动
|
||||
cmake .. \
|
||||
-DWITH_GPU=${WITH_GPU} \
|
||||
-DWITH_MKL=${WITH_MKL} \
|
||||
-DWITH_TENSORRT=${WITH_TENSORRT} \
|
||||
-DTENSORRT_LIB_DIR=${TENSORRT_LIB_DIR} \
|
||||
-DTENSORRT_INC_DIR=${TENSORRT_INC_DIR} \
|
||||
-DPADDLE_DIR=${PADDLE_DIR} \
|
||||
-DCUDA_LIB=${CUDA_LIB} \
|
||||
-DCUDNN_LIB=${CUDNN_LIB} \
|
||||
-DOPENCV_DIR=${OPENCV_DIR} \
|
||||
-DPADDLE_LIB_NAME=${PADDLE_LIB_NAME} \
|
||||
-DWITH_KEYPOINT=${WITH_KEYPOINT}
|
||||
make
|
||||
|
||||
```
|
||||
|
||||
修改脚本设置好主要参数后,执行`build`脚本:
|
||||
```shell
|
||||
sh ./scripts/build.sh
|
||||
```
|
||||
|
||||
**注意**: OPENCV依赖OPENBLAS,Ubuntu用户需确认系统是否已存在`libopenblas.so`。如未安装,可执行apt-get install libopenblas-dev进行安装。
|
||||
|
||||
### Step4: 预测及可视化
|
||||
编译成功后,预测入口程序为`build/main`其主要命令参数说明如下:
|
||||
| 参数 | 说明 |
|
||||
| ---- | ---- |
|
||||
| --model_dir | 导出的检测预测模型所在路径 |
|
||||
| --model_dir_keypoint | Option | 导出的关键点预测模型所在路径 |
|
||||
| --image_file | 要预测的图片文件路径 |
|
||||
| --image_dir | 要预测的图片文件夹路径 |
|
||||
| --video_file | 要预测的视频文件路径 |
|
||||
| --camera_id | Option | 用来预测的摄像头ID,默认为-1(表示不使用摄像头预测)|
|
||||
| --device | 运行时的设备,可选择`CPU/GPU/XPU`,默认为`CPU`|
|
||||
| --gpu_id | 指定进行推理的GPU device id(默认值为0)|
|
||||
| --run_mode | 使用GPU时,默认为paddle, 可选(paddle/trt_fp32/trt_fp16/trt_int8)|
|
||||
| --batch_size | 检测模型预测时的batch size,在指定`image_dir`时有效 |
|
||||
| --batch_size_keypoint | 关键点模型预测时的batch size,默认为8 |
|
||||
| --run_benchmark | 是否重复预测来进行benchmark测速 |
|
||||
| --output_dir | 输出图片所在的文件夹, 默认为output |
|
||||
| --use_mkldnn | CPU预测中是否开启MKLDNN加速 |
|
||||
| --cpu_threads | 设置cpu线程数,默认为1 |
|
||||
| --use_dark | 关键点模型输出预测是否使用DarkPose后处理,默认为true |
|
||||
|
||||
**注意**:
|
||||
- 优先级顺序:`camera_id` > `video_file` > `image_dir` > `image_file`。
|
||||
- --run_benchmark如果设置为True,则需要安装依赖`pip install pynvml psutil GPUtil`。
|
||||
|
||||
`样例一`:
|
||||
```shell
|
||||
#不使用`GPU`测试图片 `/root/projects/images/test.jpeg`
|
||||
./build/main --model_dir=/root/projects/models/yolov3_darknet --image_file=/root/projects/images/test.jpeg
|
||||
```
|
||||
|
||||
图片文件`可视化预测结果`会保存在当前目录下`output.jpg`文件中。
|
||||
|
||||
|
||||
`样例二`:
|
||||
```shell
|
||||
#使用 `GPU`预测视频`/root/projects/videos/test.mp4`
|
||||
./build/main --model_dir=/root/projects/models/yolov3_darknet --video_file=/root/projects/images/test.mp4 --device=GPU
|
||||
```
|
||||
视频文件目前支持`.mp4`格式的预测,`可视化预测结果`会保存在当前目录下`output.mp4`文件中。
|
||||
|
||||
|
||||
`样例三`:
|
||||
```shell
|
||||
#使用关键点模型与检测模型联合预测,使用 `GPU`预测
|
||||
#检测模型检测到的人送入关键点模型进行关键点预测
|
||||
./build/main --model_dir=/root/projects/models/yolov3_darknet --model_dir_keypoint=/root/projects/models/hrnet_w32_256x192 --image_file=/root/projects/images/test.jpeg --device=GPU
|
||||
```
|
||||
|
||||
## 性能测试
|
||||
benchmark请查看[BENCHMARK_INFER](../../BENCHMARK_INFER.md)
|
||||
158
paddle_detection/deploy/cpp/docs/windows_vs2019_build.md
Normal file
158
paddle_detection/deploy/cpp/docs/windows_vs2019_build.md
Normal file
@@ -0,0 +1,158 @@
|
||||
# Visual Studio 2019 Community CMake 编译指南
|
||||
|
||||
Windows 平台下,我们使用`Visual Studio 2019 Community` 进行了测试。微软从`Visual Studio 2017`开始即支持直接管理`CMake`跨平台编译项目,但是直到`2019`才提供了稳定和完全的支持,所以如果你想使用CMake管理项目编译构建,我们推荐你使用`Visual Studio 2019`环境下构建。
|
||||
|
||||
|
||||
## 前置条件
|
||||
* Visual Studio 2019 (根据Paddle预测库所使用的VS版本选择,请参考 [Visual Studio 不同版本二进制兼容性](https://docs.microsoft.com/zh-cn/cpp/porting/binary-compat-2015-2017?view=vs-2019) )
|
||||
* CUDA 9.0 / CUDA 10.0,cudnn 7+ / TensorRT(仅在使用GPU版本的预测库时需要)
|
||||
* CMake 3.0+ [CMake下载](https://cmake.org/download/)
|
||||
|
||||
**特别注意:windows下预测库需要的TensorRT版本为:**。
|
||||
|
||||
| 预测库版本 | TensorRT版本 |
|
||||
| ---- | ---- |
|
||||
| cuda10.1_cudnn7.6_avx_mkl_trt6 | TensorRT-6.0.1.5 |
|
||||
| cuda10.2_cudnn7.6_avx_mkl_trt7 | TensorRT-7.0.0.11 |
|
||||
| cuda11.0_cudnn8.0_avx_mkl_trt7 | TensorRT-7.2.1.6 |
|
||||
|
||||
请确保系统已经安装好上述基本软件,我们使用的是`VS2019`的社区版。
|
||||
|
||||
**下面所有示例以工作目录为 `D:\projects`演示**。
|
||||
|
||||
### Step1: 下载代码
|
||||
|
||||
下载源代码
|
||||
```shell
|
||||
git clone https://github.com/PaddlePaddle/PaddleDetection.git
|
||||
```
|
||||
|
||||
**说明**:其中`C++`预测代码在`PaddleDetection/deploy/cpp` 目录,该目录不依赖任何`PaddleDetection`下其他目录。
|
||||
|
||||
|
||||
### Step2: 下载PaddlePaddle C++ 预测库 paddle_inference
|
||||
|
||||
PaddlePaddle C++ 预测库针对不同的`CPU`和`CUDA`版本提供了不同的预编译版本,请根据实际情况下载: [C++预测库下载列表](https://paddleinference.paddlepaddle.org.cn/user_guides/download_lib.html#windows)
|
||||
|
||||
解压后`D:\projects\paddle_inference`目录包含内容为:
|
||||
```
|
||||
paddle_inference
|
||||
├── paddle # paddle核心库和头文件
|
||||
|
|
||||
├── third_party # 第三方依赖库和头文件
|
||||
|
|
||||
└── version.txt # 版本和编译信息
|
||||
```
|
||||
|
||||
### Step3: 安装配置OpenCV
|
||||
|
||||
1. 在OpenCV官网下载适用于Windows平台的3.4.6版本, [下载地址](https://sourceforge.net/projects/opencvlibrary/files/3.4.6/opencv-3.4.6-vc14_vc15.exe/download)
|
||||
2. 运行下载的可执行文件,将OpenCV解压至指定目录,如`D:\projects\opencv`
|
||||
3. 配置环境变量,如下流程所示(如果使用全局绝对路径,可以不用设置环境变量)
|
||||
- 我的电脑->属性->高级系统设置->环境变量
|
||||
- 在系统变量中找到Path(如没有,自行创建),并双击编辑
|
||||
- 新建,将opencv路径填入并保存,如`D:\projects\opencv\build\x64\vc14\bin`
|
||||
|
||||
### Step4: 编译
|
||||
|
||||
1. 进入到`cpp`文件夹
|
||||
```
|
||||
cd D:\projects\PaddleDetection\deploy\cpp
|
||||
```
|
||||
|
||||
2. 使用CMake生成项目文件
|
||||
|
||||
编译参数的含义说明如下(带`*`表示仅在使用**GPU版本**预测库时指定, 其中CUDA库版本尽量对齐,**使用9.0、10.0版本,不使用9.2、10.1等版本CUDA库**):
|
||||
|
||||
| 参数名 | 含义 |
|
||||
| ---- | ---- |
|
||||
| *CUDA_LIB | CUDA的库路径 |
|
||||
| *CUDNN_LIB | CUDNN的库路径 |
|
||||
| OPENCV_DIR | OpenCV的安装路径, |
|
||||
| PADDLE_DIR | Paddle预测库的路径 |
|
||||
| PADDLE_LIB_NAME | Paddle 预测库名称 |
|
||||
|
||||
**注意:**
|
||||
|
||||
1. 如果编译环境为CPU,需要下载`CPU`版预测库,请把`WITH_GPU`的勾去掉
|
||||
2. 如果使用的是`openblas`版本,请把`WITH_MKL`勾去掉
|
||||
3. 如无需使用关键点模型可以把`WITH_KEYPOINT`勾去掉
|
||||
4. Windows环境下,`PADDLE_LIB_NAME`需要设置为`paddle_inference`
|
||||
|
||||
执行如下命令项目文件:
|
||||
```
|
||||
cmake . -G "Visual Studio 16 2019" -A x64 -T host=x64 -DWITH_GPU=ON -DWITH_MKL=ON -DCMAKE_BUILD_TYPE=Release -DCUDA_LIB=path_to_cuda_lib -DCUDNN_LIB=path_to_cudnn_lib -DPADDLE_DIR=path_to_paddle_lib -DPADDLE_LIB_NAME=paddle_inference -DOPENCV_DIR=path_to_opencv -DWITH_KEYPOINT=ON
|
||||
```
|
||||
|
||||
例如:
|
||||
```
|
||||
cmake . -G "Visual Studio 16 2019" -A x64 -T host=x64 -DWITH_GPU=ON -DWITH_MKL=ON -DCMAKE_BUILD_TYPE=Release -DCUDA_LIB=D:\projects\packages\cuda10_0\lib\x64 -DCUDNN_LIB=D:\projects\packages\cuda10_0\lib\x64 -DPADDLE_DIR=D:\projects\packages\paddle_inference -DPADDLE_LIB_NAME=paddle_inference -DOPENCV_DIR=D:\projects\packages\opencv3_4_6 -DWITH_KEYPOINT=ON
|
||||
```
|
||||
|
||||
|
||||
|
||||
3. 编译
|
||||
用`Visual Studio 16 2019`打开`cpp`文件夹下的`PaddleObjectDetector.sln`,将编译模式设置为`Release`,点击`生成`->`全部生成
|
||||
|
||||
|
||||
### Step5: 预测及可视化
|
||||
|
||||
上述`Visual Studio 2019`编译产出的可执行文件在`out\build\x64-Release`目录下,打开`cmd`,并切换到该目录:
|
||||
|
||||
```
|
||||
cd D:\projects\PaddleDetection\deploy\cpp\out\build\x64-Release
|
||||
```
|
||||
可执行文件`main`即为样例的预测程序,其主要的命令行参数如下:
|
||||
|
||||
| 参数 | 说明 |
|
||||
| ---- | ---- |
|
||||
| --model_dir | 导出的检测预测模型所在路径 |
|
||||
| --model_dir_keypoint | Option | 导出的关键点预测模型所在路径 |
|
||||
| --image_file | 要预测的图片文件路径 |
|
||||
| --image_dir | 要预测的图片文件夹路径 |
|
||||
| --video_file | 要预测的视频文件路径 |
|
||||
| --camera_id | Option | 用来预测的摄像头ID,默认为-1(表示不使用摄像头预测)|
|
||||
| --device | 运行时的设备,可选择`CPU/GPU/XPU`,默认为`CPU`|
|
||||
| --gpu_id | 指定进行推理的GPU device id(默认值为0)|
|
||||
| --run_mode | 使用GPU时,默认为paddle, 可选(paddle/trt_fp32/trt_fp16/trt_int8)|
|
||||
| --batch_size | 检测模型预测时的batch size,在指定`image_dir`时有效 |
|
||||
| --batch_size_keypoint | 关键点模型预测时的batch size,默认为8 |
|
||||
| --run_benchmark | 是否重复预测来进行benchmark测速 |
|
||||
| --output_dir | 输出图片所在的文件夹, 默认为output |
|
||||
| --use_mkldnn | CPU预测中是否开启MKLDNN加速 |
|
||||
| --cpu_threads | 设置cpu线程数,默认为1 |
|
||||
| --use_dark | 关键点模型输出预测是否使用DarkPose后处理,默认为true |
|
||||
|
||||
**注意**:
|
||||
(1)优先级顺序:`camera_id` > `video_file` > `image_dir` > `image_file`。
|
||||
(2)如果提示找不到`opencv_world346.dll`,把`D:\projects\packages\opencv3_4_6\build\x64\vc14\bin`文件夹下的`opencv_world346.dll`拷贝到`main.exe`文件夹下即可。
|
||||
(3)--run_benchmark如果设置为True,则需要安装依赖`pip install pynvml psutil GPUtil`。
|
||||
|
||||
|
||||
`样例一`:
|
||||
```shell
|
||||
#不使用`GPU`测试图片 `D:\\images\\test.jpeg`
|
||||
.\main --model_dir=D:\\models\\yolov3_darknet --image_file=D:\\images\\test.jpeg
|
||||
```
|
||||
|
||||
图片文件`可视化预测结果`会保存在当前目录下`output.jpg`文件中。
|
||||
|
||||
|
||||
`样例二`:
|
||||
```shell
|
||||
#使用`GPU`测试视频 `D:\\videos\\test.mp4`
|
||||
.\main --model_dir=D:\\models\\yolov3_darknet --video_path=D:\\videos\\test.mp4 --device=GPU
|
||||
```
|
||||
|
||||
视频文件目前支持`.mp4`格式的预测,`可视化预测结果`会保存在当前目录下`output.mp4`文件中。
|
||||
|
||||
|
||||
`样例三`:
|
||||
```shell
|
||||
#使用关键点模型与检测模型联合预测,使用 `GPU`预测
|
||||
#检测模型检测到的人送入关键点模型进行关键点预测
|
||||
.\main --model_dir=D:\\models\\yolov3_darknet --model_dir_keypoint=D:\\models\\hrnet_w32_256x192 --image_file=D:\\images\\test.jpeg --device=GPU
|
||||
```
|
||||
|
||||
## 性能测试
|
||||
Benchmark请查看[BENCHMARK_INFER](../../BENCHMARK_INFER.md)
|
||||
Reference in New Issue
Block a user