Running mmdeploy benchmark.

This commit is contained in:
Daniel
2024-12-18 11:51:05 +01:00
parent 21e93661f5
commit 6e8f6a22ba
3 changed files with 23 additions and 1 deletions

View File

@ -46,3 +46,22 @@ mv /mmdeploy/work_dir/end2end.onnx /RapidPoseTriangulation/extras/mmdeploy/expor
python3 /RapidPoseTriangulation/extras/mmdeploy/make_extra_graphs.py
python3 /RapidPoseTriangulation/extras/mmdeploy/add_extra_steps.py
```
<br>
## Benchmark
```bash
export withFP16="_fp16"
cd /mmdeploy/
python3 ./tools/profiler.py \
configs/mmpose/pose-detection_simcc_onnxruntime_static-384x288$withFP16.py \
/mmpose/projects/rtmpose/rtmpose/body_2d_keypoint/rtmpose-m_8xb256-420e_coco-384x288.py \
/RapidPoseTriangulation/extras/mmdeploy/testimages/ \
--model /RapidPoseTriangulation/extras/mmdeploy/exports/rtmpose-m_384x288$withFP16.onnx \
--shape 384x288 \
--device cuda \
--warmup 50 \
--num-iter 200
```