MaochengHu 576cda45b8 first commit | 2 tahun lalu | |
---|---|---|
.. | ||
configs | 2 tahun lalu | |
README.md | 2 tahun lalu | |
prepare.sh | 2 tahun lalu | |
run_all.sh | 2 tahun lalu | |
run_benchmark.sh | 2 tahun lalu |
├── benchmark
│ ├── analysis_log.py
│ ├── prepare.sh
│ ├── README.md
│ ├── run_all.sh
│ ├── run_benchmark.sh
相关数据准备脚本,完成数据、模型的自动下载
主要运行脚本,可完成所有相关模型的测试方案
单模型运行脚本,可完成指定模型的测试方案
git clone https://github.com/PaddlePaddle/PaddleDetection.git
cd PaddleDetection
bash benchmark/run_all.sh
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