Skip to content

Add ONNX detector runtime pipeline - #14

Merged
Complexity-ML merged 2 commits into
Complexity-ML:mainfrom
milliyin:onnx-detector-runtime
Aug 26, 2026
Merged

Complexity-ML merged 2 commits into
Complexity-ML:mainfrom
milliyin:onnx-detector-runtime

Conversation

@milliyin

@milliyin milliyin commented Aug 25, 2026 •

Copy link
Copy Markdown
Contributor

Summary

Add a reference ONNX Runtime deployment path for TR-Hash Vision v8 detector exports.

This adds a small complexity.deploy.onnx_detector package plus a CLI example that runs the exported raw ONNX tensors through the full deployment contract:

  • image preprocessing and letterbox restoration
  • ONNX Runtime inference
  • shared DFL box decoding
  • confidence filtering for both branches
  • class-aware NMS for O2M only
  • NMS-free top-score filtering without NMS
  • stable JSON detection schema with normalized boxes, pixel boxes, class IDs, and scores
  • provider selection/reporting for CPU, CUDA, and TensorRT

Docs

Adds:

  • docs/onnx_deploy_contract.md
  • docs/onnx_deploy.md

The docs include the raw tensor contract, grid/DFL decode rules, branch behavior, and runnable CPU/CUDA CLI examples.

Testing

Passed:

python -m pytest -q tests\test_onnx_detector_skeleton.py tests\test_onnx_detector_core.py tests\test_onnx_detector_metadata.py tests\test_onnx_detector_pipeline.py tests\test_onnx_detect_cli.py
16 passed

Issue closed #13

@Complexity-ML
Complexity-ML merged commit 03d3425 into Complexity-ML:main Aug 26, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add a reference ONNX post-processing pipeline for both Vision v8 branches

2 participants