Exporting onnx instructions.

2024-11-29 13:51:20 +01:00
parent 1c0eeb9147
commit 4b2811c1c8
6 changed files with 97 additions and 0 deletions
--- a/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288.py
+++ b/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288.py
@ -0,0 +1,8 @@
+_base_ = ["./pose-detection_static.py", "../_base_/backends/onnxruntime.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