shouxieai/tensorRT_Pro
C++/Python library providing high-performance inference using NVIDIA TensorRT for object detection models like YOLO variants.

Velocity · 7d
+1.6
★ / day
Trend
→steady
star history
TensorRT_Pro is a framework for optimized model inference using NVIDIA TensorRT. It converts PyTorch models to TensorRT engines and provides simple C++ and Python APIs for deployment. The library supports YOLOv5, YOLOX, YOLOv7, and other object detection architectures, handling tasks such as model serialization, GPU inference, and post-processing with hardware decoding integration.