Export pose postprocessing steps as well.

2024-11-29 17:59:15 +01:00
parent 19015c6326
commit ad1ef42b99
3 changed files with 7 additions and 13 deletions
--- a/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288.py
+++ b/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288.py
@ -2,7 +2,7 @@ _base_ = ["./pose-detection_static.py", "../_base_/backends/onnxruntime.py"]

 onnx_config = dict(
    input_shape=[288, 384],
-    output_names=["simcc_x", "simcc_y"],
+    output_names=["kpts", "scores"],
 )

-codebase_config = dict(export_postprocess=False)  # do not export get_simcc_maximum
+codebase_config = dict(export_postprocess=True)  # export get_simcc_maximum