Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleDetection
提交
f5d8f770
P
PaddleDetection
项目概览
PaddlePaddle
/
PaddleDetection
接近 2 年 前同步成功
通知
707
Star
11112
Fork
2696
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
184
列表
看板
标记
里程碑
合并请求
40
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
184
Issue
184
列表
看板
标记
里程碑
合并请求
40
合并请求
40
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
f5d8f770
编写于
12月 27, 2021
作者:
W
wangguanzhong
提交者:
GitHub
12月 27, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
support windows deploy for pptracking (#4969)
上级
ebbe6686
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
39 addition
and
6 deletion
+39
-6
deploy/pptracking/cpp/README.md
deploy/pptracking/cpp/README.md
+33
-1
deploy/pptracking/cpp/src/jde_predictor.cc
deploy/pptracking/cpp/src/jde_predictor.cc
+3
-2
deploy/pptracking/cpp/src/tracker.cc
deploy/pptracking/cpp/src/tracker.cc
+3
-3
未找到文件。
deploy/pptracking/cpp/README.md
浏览文件 @
f5d8f770
...
...
@@ -45,6 +45,9 @@ paddle_inference
### 3. 编译
<details>
<summary>
Linux编译:
</summary>
编译
`cmake`
的命令在
`scripts/build.sh`
中,请根据实际情况修改主要参数,其主要内容说明如下:
```
...
...
@@ -85,10 +88,39 @@ OPENCV_DIR=/path/to/opencv
sh ./scripts/build.sh
```
</details>
<details>
<summary>
Windows编译:
</summary>
-
安装配置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`
-
使用CMake生成项目文件
执行如下命令项目文件:
```
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
```
-
编译
用
`Visual Studio 2019`
打开
`cpp`
文件夹下的
`PaddleObjectDetector.sln`
,将编译模式设置为
`Release`
,点击
`生成`
->
`全部生成
编译产出的可执行文件在`
Release
`目录下
</details>
**注意:**
1. `
TX2
`平台的`
CUDA
`、`
CUDNN
`需要通过`
JetPack
`安装。
2. 已提供linux和tx2平台的opencv下载方式,其他环境请自行安装[opencv](https://opencv.org/)
3. Windows用户推荐使用Visual Studio 2019编译
## 二、导出预测模型
...
...
@@ -104,7 +136,7 @@ python tools/export_model.py -c configs/mot/fairmot/fairmot_hrnetv2_w18_dlafpn_3
## 三、C++预测
完成以上步骤后,可以通过
```build/main```
进行预测,参数列表如下:
完成以上步骤后,可以通过```build/main```
(Linux)或```main.exe```(Windows)
进行预测,参数列表如下:
| 参数 | 说明 |
| ---- | ---- |
...
...
deploy/pptracking/cpp/src/jde_predictor.cc
浏览文件 @
f5d8f770
...
...
@@ -32,7 +32,7 @@ void JDEPredictor::LoadModel(const std::string& model_dir,
config
.
EnableUseGpu
(
200
,
this
->
gpu_id_
);
config
.
SwitchIrOptim
(
true
);
// use tensorrt
if
(
run_mode
!=
"
fluid
"
)
{
if
(
run_mode
!=
"
paddle
"
)
{
auto
precision
=
paddle_infer
::
Config
::
Precision
::
kFloat32
;
if
(
run_mode
==
"trt_fp32"
)
{
precision
=
paddle_infer
::
Config
::
Precision
::
kFloat32
;
...
...
@@ -42,7 +42,8 @@ void JDEPredictor::LoadModel(const std::string& model_dir,
precision
=
paddle_infer
::
Config
::
Precision
::
kInt8
;
}
else
{
printf
(
"run_mode should be 'fluid', 'trt_fp32', 'trt_fp16' or 'trt_int8'"
);
"run_mode should be 'paddle', 'trt_fp32', 'trt_fp16' or "
"'trt_int8'"
);
}
// set tensorrt
config
.
EnableTensorRtEngine
(
1
<<
30
,
...
...
deploy/pptracking/cpp/src/tracker.cc
浏览文件 @
f5d8f770
...
...
@@ -191,9 +191,9 @@ bool JDETracker::update(const cv::Mat &dets,
tracks
->
clear
();
for
(
size_t
i
=
0
;
i
<
this
->
tracked_trajectories
.
size
();
++
i
)
{
if
(
this
->
tracked_trajectories
[
i
].
is_activated
)
{
Track
track
=
{
.
id
=
this
->
tracked_trajectories
[
i
].
id
,
.
score
=
this
->
tracked_trajectories
[
i
].
score
,
.
ltrb
=
this
->
tracked_trajectories
[
i
].
ltrb
};
Track
track
=
{
this
->
tracked_trajectories
[
i
].
id
,
this
->
tracked_trajectories
[
i
].
score
,
this
->
tracked_trajectories
[
i
].
ltrb
};
tracks
->
push_back
(
track
);
}
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录