Skip to content

Conversation

@wangzijian1010
Copy link
Member

This pull request adds support for the YOLOv11 object detection model to the lite TensorRT-based computer vision toolkit. The main changes include implementing the TRTYOLOV11 class for inference, updating the build system, and adding a corresponding test example. These updates allow users to run YOLOv11 inference with proper preprocessing, postprocessing, and NMS, similar to other YOLO models in the toolkit.

YOLOv11 Model Integration

  • Added implementation of the TRTYOLOV11 class in trt_yolov11.cpp and its header, including preprocessing (letterbox resizing, normalization), inference, and postprocessing (bounding box decoding and NMS). [1] [2]
  • Declared the TRTYOLOV11 class in the core header and registered it in the type system for easy use as lite::trt::cv::detection::YOLOV11. [1] [2] [3] [4]

Build System and Example

  • Updated CMake configuration to add a new executable for YOLOv11 (lite_yolov11).
  • Added a new test example test_lite_yolov11.cpp to demonstrate and validate YOLOv11 inference using TensorRT.

Copy link
Member

@DefTruth DefTruth left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for you contribution! Please also update the support matrix!

Copy link
Member

@DefTruth DefTruth left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@wangzijian1010 wangzijian1010 merged commit e53c332 into xlite-dev:main Dec 12, 2025
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.

2 participants