移动paddle_detection
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
# 通用检测benchmark测试脚本说明
|
||||
|
||||
```
|
||||
├── benchmark
|
||||
│ ├── analysis_log.py
|
||||
│ ├── prepare.sh
|
||||
│ ├── README.md
|
||||
│ ├── run_all.sh
|
||||
│ ├── run_benchmark.sh
|
||||
```
|
||||
|
||||
## 脚本说明
|
||||
|
||||
### prepare.sh
|
||||
相关数据准备脚本,完成数据、模型的自动下载
|
||||
### run_all.sh
|
||||
主要运行脚本,可完成所有相关模型的测试方案
|
||||
### run_benchmark.sh
|
||||
单模型运行脚本,可完成指定模型的测试方案
|
||||
|
||||
## Docker 运行环境
|
||||
* docker image: registry.baidubce.com/paddlepaddle/paddle:2.1.2-gpu-cuda10.2-cudnn7
|
||||
* paddle = 2.1.2
|
||||
* python = 3.7
|
||||
|
||||
## 运行benchmark测试
|
||||
|
||||
### 运行所有模型
|
||||
```
|
||||
git clone https://github.com/PaddlePaddle/PaddleDetection.git
|
||||
cd PaddleDetection
|
||||
bash benchmark/run_all.sh
|
||||
```
|
||||
|
||||
### 运行指定模型
|
||||
* Usage:bash run_benchmark.sh ${run_mode} ${batch_size} ${fp_item} ${max_epoch} ${model_name}
|
||||
* model_name: faster_rcnn, fcos, deformable_detr, gfl, hrnet, higherhrnet, solov2, jde, fairmot
|
||||
```
|
||||
git clone https://github.com/PaddlePaddle/PaddleDetection.git
|
||||
cd PaddleDetection
|
||||
bash benchmark/prepare.sh
|
||||
|
||||
# 单卡
|
||||
CUDA_VISIBLE_DEVICES=0 bash benchmark/run_benchmark.sh sp 2 fp32 1 faster_rcnn
|
||||
# 多卡
|
||||
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 bash benchmark/run_benchmark.sh mp 2 fp32 1 faster_rcnn
|
||||
```
|
||||
Reference in New Issue
Block a user