Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
f5d8f770
P
PaddleDetection
项目概览
s920243400
/
PaddleDetection
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleDetection
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
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.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录