Running mmdeploy benchmark.
This commit is contained in:
@ -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
|
||||||
|
```
|
||||||
|
|||||||
@ -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
|
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"]
|
||||||
|
|||||||
BIN
extras/mmdeploy/testimages/human-pose.jpeg
Normal file
BIN
extras/mmdeploy/testimages/human-pose.jpeg
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 38 KiB |
Reference in New Issue
Block a user