Loading [MathJax]/jax/output/HTML-CSS/jax.js
72 Star 780 Fork 286

GVPPaddlePaddle/PaddleDetection

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

通用检测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 runmode{batch_size} fpitem{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
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/paddlepaddle/PaddleDetection.git
git@gitee.com:paddlepaddle/PaddleDetection.git
paddlepaddle
PaddleDetection
PaddleDetection
release/2.8.1

搜索帮助