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/make_extra_graphs.py
python3 /RapidPoseTriangulation/extras/mmdeploy/add_extra_steps.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
```

View File

@ -31,5 +31,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends python3-tk
# Tool for fp16 conversion # Tool for fp16 conversion
RUN pip3 install --upgrade --no-cache-dir onnxconverter_common RUN pip3 install --upgrade --no-cache-dir onnxconverter_common
# Fix an error when profiling
RUN pip3 install --upgrade --no-cache-dir "onnxruntime-gpu<1.17"
WORKDIR /mmdeploy/ WORKDIR /mmdeploy/
CMD ["/bin/bash"] CMD ["/bin/bash"]

Binary file not shown.

After

Width:  |  Height:  |  Size: 38 KiB