๐ Summary
This release, `v8.3.85`, focuses on improving TensorRT export functionality and refining ONNX segmentation examples for better performance and usability. ๐
---
๐ Key Changes
TensorRT Updates (Main Focus of Release):
- **Fixed `max_shape` Calculation Bug**: Resolved inconsistent calculations during TensorRT export with non-zero workspace values.
- **Improved Default Behavior**: Ensured workspace settings default to `0` when not specified.
ONNX Segmentation Example Enhancements:
- **Streamlined Preprocessing/Postprocessing**: Simplified the example to use a more efficient and flexible implementation, including key parameters like `iou`, `imgsz`, and `conf`.
- **Unified Confidence Threshold**: Updated ONNX example to align with YOLO conventions for clarity (`conf`, `iou`).
- **Optimized Mask Handling**: Enhanced segmentation mask functionality for better accuracy and resource efficiency.
- **Adaptive Backend Setup**: Integrated GPU support automatically when available, improving ease of use.
---
๐ฏ Purpose & Impact
- **For TensorRT Users**:
- **Purpose**: Addressed critical bugs in TensorRT export needed for high-performance inference engines.
- **Impact**: Accurate shape calculation ensures smoother deployments of dynamic models with TensorRT, preventing crashes or incorrect behavior. Users exporting YOLO models into `.engine` format, especially with non-zero workspaces, will experience stable exports.
- **For ONNX Developers**:
- **Purpose**: Simplify, optimize, and enhance usability of segmentation examples using ONNX Runtime.
- **Impact**: Developers using ONNX Runtime benefit from faster setup, more intuitive parameter usage, and reliable mask-based object segmentation results.
- **General Improvements**:
- These updates improve the overall developer experience and model deployment across various inference backends, enabling seamless workflows for cutting-edge scenarios.
---
This update primarily strengthens export and inference capabilities ๐ฏ, while making the process more robust for advanced and typical users alike.
What's Changed
* Cleanup and fix ONNX segment example by Y-T-G in https://github.com/ultralytics/ultralytics/pull/19551
* `ultralytics 8.3.85` TensoRT export `max_shape` fix by Y-T-G in https://github.com/ultralytics/ultralytics/pull/19541
**Full Changelog**: https://github.com/ultralytics/ultralytics/compare/v8.3.84...v8.3.85