diff --git a/extras/mmdeploy/README.md b/extras/mmdeploy/README.md index 91be721..84dbdd4 100644 --- a/extras/mmdeploy/README.md +++ b/extras/mmdeploy/README.md @@ -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 ``` + +
+ +## 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 +``` diff --git a/extras/mmdeploy/dockerfile b/extras/mmdeploy/dockerfile index 7eae439..b285ba6 100644 --- a/extras/mmdeploy/dockerfile +++ b/extras/mmdeploy/dockerfile @@ -29,7 +29,10 @@ RUN echo 'export LD_LIBRARY_PATH=/mmdeploy/../mmdeploy-dep/onnxruntime-linux-x64 RUN apt-get update && apt-get install -y --no-install-recommends python3-tk # 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/ CMD ["/bin/bash"] diff --git a/extras/mmdeploy/testimages/human-pose.jpeg b/extras/mmdeploy/testimages/human-pose.jpeg new file mode 100644 index 0000000..8de4015 Binary files /dev/null and b/extras/mmdeploy/testimages/human-pose.jpeg differ