Repository navigation
Add TensorRT deployment and latency benchmarks - #4
Merged
Merged
Conversation
Closed
Maelic
marked this pull request as ready for review
September 23, 2026 04:47
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Adds native TensorRT 10 deployment for the relation model, addressing #2. Users can build local FP32 engines and run the image/webcam demo with
--backend tensorrt, or useTensorRTRelationHeadwith their own boxes. Vocabulary selection, calibration and spatial/semantic decoding share the ONNX implementation. GPU dependencies remain optional.The README leads with a runnable example, model selection and deployment choices, uses Figure 2 from arXiv:2609.12552v1 with attribution, and keeps just two generated latency highlights for the recommended model. Full runtime/detector comparisons, hardware conditions, p95 and methodology live in the linked benchmark docs. Research and dataset details also link to dedicated pages.
Completed GPU comparison
--source. Earlier negative findings remain documented in Markdown.Sparse-input correction
The detector preflight found duplicate valid pairs in ONNX Runtime CUDA when the sampler padded with the dtype minimum. Changing the masked selection sentinel fixes the sparse-input case without changing learned weights. A CUDA regression test covers empty, singleton, sparse and dense inputs. All three ONNX graphs and TensorRT engines used for the new measurements were rebuilt from the released EMA checkpoints. The failed preflight is documented, with its diagnostic output retained locally, and contributes no timing samples.
Validation
atol=1e-3,rtol=1e-4).atol=1e-2,rtol=1e-4).--checkagainst the retained local results; local links, formatting andgit diff --checkpass.TensorRT scope: FP32 with TF32 disabled; engines are built on the target GPU. Jetson and reduced-precision TensorRT remain unvalidated. No detector weights are redistributed.
Closes #2.