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

Moved detector pre/post-processing into onnx graph.

Browse Source
This commit is contained in:
Daniel
2024-12-03 11:52:55 +01:00
parent 36781e616b
commit 742d2386c7
5 changed files with 213 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extras/mmdeploy/configs/detection_onnxruntime_static-320x320.py
View File

@ -3,3 +3,7 @@ _base_ = ["../_base_/base_static.py", "../../_base_/backends/onnxruntime.py"]
onnx_config = dict(
input_shape=[320, 320],
)
codebase_config = dict(
post_processing=dict(score_threshold=0.3, iou_threshold=0.3),
)

4
extras/mmdeploy/configs/detection_onnxruntime_static-320x320_fp16.py
View File

@ -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),
)