Moved detector pre/post-processing into onnx graph.

2024-12-03 11:52:55 +01:00
parent 36781e616b
commit 742d2386c7
5 changed files with 213 additions and 104 deletions
--- a/extras/mmdeploy/configs/detection_onnxruntime_static-320x320_fp16.py
+++ b/extras/mmdeploy/configs/detection_onnxruntime_static-320x320_fp16.py
@ -3,3 +3,7 @@ _base_ = ["../_base_/base_static.py", "../../_base_/backends/onnxruntime-fp16.py
 onnx_config = dict(
    input_shape=[320, 320],
 )
+
+codebase_config = dict(
+    post_processing=dict(score_threshold=0.3, iou_threshold=0.3),
+)