feat(export): add YOLO26 support and refactor trtyolo-export#293
Merged
Conversation
- replace the legacy framework-specific export flow with an ONNX-first conversion pipeline that rewrites exported models into TensorRT-YOLO compatible outputs - split the implementation into dedicated cli, pipeline, model_io, plugin, handler, logging, type, and utility modules so the conversion flow is easier to extend and maintain - add handler abstractions and model-specific graph rewrite handlers for classify, legacy Ultralytics heads, and nms-free output patterns - remove the old head-based and PP-YOLOE-specific export path in favor of a unified rewrite flow driven by graph inspection and TensorRT plugin registration - expose a cleaner package entrypoint with __main__ support and package-level exports for CLI and future library-style usage
![sourcery-ai[bot]](https://avatars.githubusercontent.com/in/48477?s=60&v=4){kind=small}
This comment was marked as resolved.
This comment was marked as resolved.
Sorry, something went wrong.
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
1 participant
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.
Summary
Closes #285.
This PR adds YOLO26 support to
trtyolo-export. To make that support maintainable, it also reworks the exporter froma monolithic implementation into a modular ONNX conversion pipeline.
What changed
trtyolo-exportinto focused modules for pipeline orchestration, handlers, model I/O, plugin injection,logging, and shared utilities
README.mdandREADME.cn.mdto document the v2 export workflow and YOLO26 support matrixWhy
Supporting YOLO26 cleanly in the previous exporter structure would keep increasing the complexity of the monolithic
conversion logic. This rework isolates model-family-specific rewrite logic, makes YOLO26 support easier to maintain,
and provides a cleaner base for future exporter updates.