crosstyan/RapidPoseTriangulation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Exporting fp16 onnx models.

Browse Source
This commit is contained in:
Daniel
2024-11-29 17:37:44 +01:00
parent 7b9505ca02
commit 19015c6326
10 changed files with 107 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extras/mmdeploy/configs/detection_onnxruntime_static-320x320_fp16.py Normal file
View File

@ -0,0 +1,5 @@
_base_ = ["../_base_/base_static.py", "../../_base_/backends/onnxruntime-fp16.py"]
onnx_config = dict(
input_shape=[320, 320],
)

8
extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288_fp16.py Normal file
View File

@ -0,0 +1,8 @@
_base_ = ["./pose-detection_static.py", "../_base_/backends/onnxruntime-fp16.py"]
onnx_config = dict(
input_shape=[288, 384],
output_names=["simcc_x", "simcc_y"],
)
codebase_config = dict(export_postprocess=False) # do not export get_simcc_maximum