Skip to content

Commit 6ddca83

Browse files
authored
update ppvehicle modelzoo (PaddlePaddle#6679)
* update ppvehicle modelzoo; test=document_fix * update pipeline pictures * update * fix platerec jetson trtfp16; test=document_fix * replace all --model_dir in pipeline docs; test=document_fix
1 parent aa5db28 commit 6ddca83

15 files changed

+45
-49
lines changed

deploy/pipeline/README.md

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@
6262
| 打电话识别 | 单人ms | [目标检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip)<br>[基于人体id的图像分类](https://bj.bcebos.com/v1/paddledet/models/pipeline/PPHGNet_tiny_calling_halfbody.zip) | 目标检测:182M<br>基于人体id的图像分类:45M |
6363

6464

65-
点击模型方案中的模型即可下载指定模型,下载后解压存放至`./output_inference`目录中
65+
点击模型方案中的模型即可下载指定模型,下载后解压存放至`./output_inference`目录中
6666

6767
</details>
6868

@@ -71,20 +71,14 @@
7171
<details>
7272
<summary><b>端到端模型效果(点击展开)</b></summary>
7373

74-
| 任务 | 端到端速度(ms) | 模型方案 | 模型体积 |
75-
| --------- | --------- | --------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | ------------------------------------------- |
76-
| 行人检测(高精度) | 25.1ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip) | 182M |
77-
| 行人检测(轻量级) | 16.2ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_s_36e_pipeline.zip) | 27M |
78-
| 行人跟踪(高精度) | 31.8ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip) | 182M |
79-
| 行人跟踪(轻量级) | 21.0ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_s_36e_pipeline.zip) | 27M |
80-
| 属性识别(高精度) | 单人8.5ms | [目标检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip)<br> [属性识别](https://bj.bcebos.com/v1/paddledet/models/pipeline/strongbaseline_r50_30e_pa100k.zip) | 目标检测:182M<br>属性识别:86M |
81-
| 属性识别(轻量级) | 单人7.1ms | [目标检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip)<br> [属性识别](https://bj.bcebos.com/v1/paddledet/models/pipeline/strongbaseline_r50_30e_pa100k.zip) | 目标检测:182M<br>属性识别:86M |
82-
| 摔倒识别 | 单人10ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip) <br> [关键点检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/dark_hrnet_w32_256x192.zip) <br> [基于关键点行为识别](https://bj.bcebos.com/v1/paddledet/models/pipeline/STGCN.zip) | 多目标跟踪:182M<br>关键点检测:101M<br>基于关键点行为识别:21.8M |
83-
| 闯入识别 | 31.8ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip) | 182M |
84-
| 打架识别 | 19.7ms | [视频分类](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip) | 90M |
85-
| 抽烟识别 | 单人15.1ms | [目标检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip)<br>[基于人体id的目标检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/ppyoloe_crn_s_80e_smoking_visdrone.zip) | 目标检测:182M<br>基于人体id的目标检测:27M |
86-
| 打电话识别 | 单人ms | [目标检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip)<br>[基于人体id的图像分类](https://bj.bcebos.com/v1/paddledet/models/pipeline/PPHGNet_tiny_calling_halfbody.zip) | 目标检测:182M<br>基于人体id的图像分类:45M |
87-
74+
| 任务 | 端到端速度(ms)| 模型方案 | 模型体积 |
75+
| :---------: | :-------: | :------: |:------: |
76+
| 车辆检测(高精度) | 25.7ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip) | 182M |
77+
| 车辆检测(轻量级) | 13.2ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_s_36e_ppvehicle.zip) | 27M |
78+
| 车辆跟踪(高精度) | 40ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip) | 182M |
79+
| 车辆跟踪(轻量级) | 25ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_s_36e_ppvehicle.zip) | 27M |
80+
| 车牌识别 | 4.68ms | [车牌检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/ch_PP-OCRv3_det_infer.tar.gz) <br> [车牌字符识别](https://bj.bcebos.com/v1/paddledet/models/pipeline/ch_PP-OCRv3_rec_infer.tar.gz) | 车牌检测:3.9M <br> 车牌字符识别: 12M |
81+
| 车辆属性 | 7.31ms | [车辆属性](https://bj.bcebos.com/v1/paddledet/models/pipeline/vehicle_attribute_model.zip) | 7.2M |
8882

8983
点击模型方案中的模型即可下载指定模型,下载后解压存放至`./output_inference`目录中
9084

deploy/pipeline/docs/tutorials/PPVehicle_QUICK_STARTED.md

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,13 @@ PP-Vehicle提供了目标检测、属性识别、行为识别、ReID预训练模
4646

4747
| 任务 | 端到端速度(ms)| 模型方案 | 模型体积 |
4848
| :---------: | :-------: | :------: |:------: |
49-
| 车辆检测(高精度) | 25.1ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip) | 182M |
50-
| 车辆检测(轻量级) | 16.2ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_s_36e_ppvehicle.zip) | 27M |
51-
| 车辆跟踪(高精度) | 31.8ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip) | 182M |
52-
| 车辆跟踪(轻量级) | 21.0ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_s_36e_ppvehicle.zip) | 27M |
53-
| 闯入识别 | 31.8ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip) | 多目标跟踪:182M |
49+
| 车辆检测(高精度) | 25.7ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip) | 182M |
50+
| 车辆检测(轻量级) | 13.2ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_s_36e_ppvehicle.zip) | 27M |
51+
| 车辆跟踪(高精度) | 40ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip) | 182M |
52+
| 车辆跟踪(轻量级) | 25ms | [多目标跟踪](https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_s_36e_ppvehicle.zip) | 27M |
53+
| 车牌识别 | 4.68ms | [车牌检测](https://bj.bcebos.com/v1/paddledet/models/pipeline/ch_PP-OCRv3_det_infer.tar.gz) <br> [车牌字符识别](https://bj.bcebos.com/v1/paddledet/models/pipeline/ch_PP-OCRv3_rec_infer.tar.gz) | 车牌检测:3.9M <br> 车牌字符识别: 12M |
54+
| 车辆属性 | 7.31ms | [车辆属性](https://bj.bcebos.com/v1/paddledet/models/pipeline/vehicle_attribute_model.zip) | 7.2M |
55+
5456

5557
下载模型后,解压至`./output_inference`文件夹。
5658

@@ -97,7 +99,7 @@ VEHICLE_ATTR:
9799
**注意:**
98100

99101
- 如果用户需要实现不同任务,可以在配置文件对应enable选项设置为True。
100-
- 如果用户仅需要修改模型文件路径,可以在命令行中加入 `--model_dir det=ppyoloe/` 即可,也可以手动修改配置文件中的相应模型路径,详细说明参考下方参数说明文档。
102+
- 如果用户仅需要修改模型文件路径,可以在命令行中--config后面紧跟着 `-o MOT.model_dir=ppyoloe/` 进行修改即可,也可以手动修改配置文件中的相应模型路径,详细说明参考下方参数说明文档。
101103

102104

103105
## 预测部署
@@ -111,7 +113,7 @@ python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppv
111113
112114
# 车辆跟踪,指定配置文件路径,模型路径和测试视频,在配置文件```deploy/pipeline/config/infer_cfg_ppvehicle.yml```中的MOT部分enable设置为```True```
113115
# 命令行中指定的模型路径优先级高于配置文件
114-
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --video_file=test_video.mp4 --device=gpu --model_dir det=ppyoloe/ [--run_mode trt_fp16]
116+
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --video_file=test_video.mp4 --device=gpu [--run_mode trt_fp16]
115117
116118
# 车辆属性识别,指定配置文件路径和测试视频,在配置文件```deploy/pipeline/config/infer_cfg_ppvehicle.yml```中的ATTR部分enable设置为```True```
117119
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --video_file=test_video.mp4 --device=gpu [--run_mode trt_fp16]
@@ -130,7 +132,6 @@ python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppv
130132
|-------|-------|----------|
131133
| --config | Yes | 配置文件路径 |
132134
| -o | Option | 覆盖配置文件中对应的配置 |
133-
| --model_dir | Option | 各任务模型路径,优先级高于配置文件, 例如`--model_dir det=better_det/ attr=better_attr/`|
134135
| --image_file | Option | 需要预测的图片 |
135136
| --image_dir | Option | 要预测的图片文件夹路径 |
136137
| --video_file | Option | 需要预测的视频,或者rtsp流地址 |

deploy/pipeline/docs/tutorials/pphuman_action.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,12 +60,12 @@ python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pph
6060
3. 若修改模型路径,有以下两种方式:
6161

6262
- ```./deploy/pipeline/config/infer_cfg_pphuman.yml```下可以配置不同模型路径,关键点模型和摔倒行为识别模型分别对应`KPT``SKELETON_ACTION`字段,修改对应字段下的路径为实际期望的路径即可。
63-
- 命令行中增加`--model_dir`修改模型路径:
63+
- 命令行中--config后面紧跟着增加`-o KPT.model_dir=xxx SKELETON_ACTION.model_dir=xxx `修改模型路径:
6464
```python
6565
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pphuman.yml \
66+
-o KPT.model_dir=./dark_hrnet_w32_256x192 SKELETON_ACTION.model_dir=./STGCN\
6667
--video_file=test_video.mp4 \
67-
--device=gpu \
68-
--model_dir kpt=./dark_hrnet_w32_256x192 action=./STGCN
68+
--device=gpu
6969
```
7070
4. 启动命令中的完整参数说明,请参考[参数说明](./PPHuman_QUICK_STARTED.md)
7171

deploy/pipeline/docs/tutorials/pphuman_action_en.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -72,13 +72,13 @@ SKELETON_ACTION: # Config for skeleton-based action recognition model
7272
3. There are two ways to modify the model path:
7373

7474
- In ```./deploy/pipeline/config/infer_cfg_pphuman.yml```, you can configurate different model paths,which is proper only if you match keypoint models and action recognition models with the fields of `KPT` and `SKELETON_ACTION` respectively, and modify the corresponding path of each field into the expected path.
75-
- Add `--model_dir` in the command line to revise the model path:
75+
- Add `-o KPT.model_dir=xxx SKELETON_ACTION.model_dir=xxx ` in the command line following the --config to change the model path:
7676

7777
```python
78-
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pphuman.yml \
79-
--video_file=test_video.mp4 \
80-
--device=gpu \
81-
--model_dir kpt=./dark_hrnet_w32_256x192 action=./STGCN
78+
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pphuman.yml \
79+
-o KPT.model_dir=./dark_hrnet_w32_256x192 SKELETON_ACTION.model_dir=./STGCN\
80+
--video_file=test_video.mp4 \
81+
--device=gpu
8282
```
8383
4. For detailed parameter description, please refer to [Parameter Description](./PPHuman_QUICK_STARTED.md)
8484

deploy/pipeline/docs/tutorials/pphuman_attribute.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -59,12 +59,12 @@ python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pph
5959
4. 若修改模型路径,有以下两种方式:
6060

6161
- 方法一:```./deploy/pipeline/config/infer_cfg_pphuman.yml```下可以配置不同模型路径,属性识别模型修改ATTR字段下配置
62-
- 方法二:命令行中增加`--model_dir`修改模型路径:
62+
- 方法二:命令行中--config后面紧跟着增加`-o ATTR.model_dir`修改模型路径:
6363
```python
64-
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pphuman.yml \
64+
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pphuman.yml
65+
-o ATTR.model_dir=output_inference/PPLCNet_x1_0_person_attribute_945_infer/\
6566
--video_file=test_video.mp4 \
66-
--device=gpu \
67-
--model_dir attr=output_inference/PPLCNet_x1_0_person_attribute_945_infer/
67+
--device=gpu
6868
```
6969

7070
测试效果如下:

deploy/pipeline/docs/tutorials/pphuman_attribute_en.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,12 +55,12 @@ python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pph
5555
4. If you want to change the model path, there are two methods:
5656

5757
- The first: In ```./deploy/pipeline/config/infer_cfg_pphuman.yml``` you can configurate different model paths. In attribute recognition models, you can modify the configuration in the field of ATTR.
58-
- The second: Add `--model_dir` in the command line to change the model path:
58+
- The second: Add `-o ATTR.model_dir` in the command line following the --config to change the model path:
5959
```python
6060
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pphuman.yml \
61+
-o ATTR.model_dir=output_inference/PPLCNet_x1_0_person_attribute_945_infer/\
6162
--video_file=test_video.mp4 \
62-
--device=gpu \
63-
--model_dir attr=output_inference/PPLCNet_x1_0_person_attribute_945_infer/
63+
--device=gpu
6464
```
6565

6666
The test result is:

deploy/pipeline/docs/tutorials/pphuman_mot.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,15 +39,15 @@ python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pph
3939
4. 若修改模型路径,有以下两种方式:
4040

4141
- ```./deploy/pipeline/config/infer_cfg_pphuman.yml```下可以配置不同模型路径,检测和跟踪模型分别对应`DET``MOT`字段,修改对应字段下的路径为实际期望的路径即可。
42-
- 命令行中增加`--model_dir`修改模型路径:
42+
- 命令行中--config后面紧跟着增加`-o MOT.model_dir`修改模型路径:
4343
```python
4444
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pphuman.yml \
45+
-o MOT.model_dir=ppyoloe/\
4546
--video_file=test_video.mp4 \
4647
--device=gpu \
4748
--region_type=horizontal \
4849
--do_entrance_counting \
49-
--draw_center_traj \
50-
--model_dir det=ppyoloe/
50+
--draw_center_traj
5151

5252
```
5353
**注意:**

deploy/pipeline/docs/tutorials/pphuman_mot_en.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,16 +39,16 @@ python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pph
3939
4. There are two ways to modify the model path:
4040

4141
- In `./deploy/pipeline/config/infer_cfg_pphuman.yml`, you can configurate different model paths,which is proper only if you match keypoint models and action recognition models with the fields of `DET` and `MOT` respectively, and modify the corresponding path of each field into the expected path.
42-
- Add `--model_dir` in the command line to revise the model path:
42+
- Add `-o MOT.model_dir` in the command line following the --config to change the model path
4343

4444
```python
4545
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pphuman.yml \
46+
-o MOT.model_dir=ppyoloe/\
4647
--video_file=test_video.mp4 \
4748
--device=gpu \
4849
--region_type=horizontal \
4950
--do_entrance_counting \
50-
--draw_center_traj \
51-
--model_dir det=ppyoloe/
51+
--draw_center_traj
5252

5353
```
5454
**Note:**

deploy/pipeline/docs/tutorials/pphuman_mtmct.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,9 @@ python3 deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pp
1818

1919
```python
2020
python3 deploy/pipeline/pipeline.py
21-
--config deploy/pipeline/config/infer_cfg_pphuman.yml
21+
--config deploy/pipeline/config/infer_cfg_pphuman.yml -o REID.model_dir=reid_best/
2222
--video_dir=[your_video_file_directory]
2323
--device=gpu
24-
--model_dir reid=reid_best/
2524
```
2625

2726
## 方案说明

deploy/pipeline/docs/tutorials/pphuman_mtmct_en.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,9 @@ python3 deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_pp
1818

1919
```python
2020
python3 deploy/pipeline/pipeline.py
21-
--config deploy/pipeline/config/infer_cfg_pphuman.yml
21+
--config deploy/pipeline/config/infer_cfg_pphuman.yml -o REID.model_dir=reid_best/
2222
--video_dir=[your_video_file_directory]
2323
--device=gpu
24-
--model_dir reid=reid_best/
2524
```
2625

2726
## Intorduction to the Solution

0 commit comments

Comments
 (0)