THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection
https://arxiv.org/abs/2405.14458
GNU Affero General Public License v3.0
8k stars 662 forks source link

End2End implementation for TensorRT - Added Support on DeepStream #283

Open levipereira opened 1 week ago

levipereira commented 1 week ago

Just sharing!!

Repository 1: YOLOv10 with Enhanced Outputs and TensorRT Optimization

YOLOv10 Export ONNX

This repository extends the YOLOv10 model with additional output layers: num_detections, detection_boxes, detection_scores, and detection_classes. It also includes optimizations for exporting models with dynamic size and dynamic batch size, ensuring that the ONNX models are optimized for TensorRT builds. image

Repository 2: DeepStream Integration for YOLOv10

DeepStream YOLOv10

This repository provides a practical implementation of YOLOv10 within DeepStream, a powerful tool for video analytics. Users can seamlessly integrate the YOLOv10 model into their projects using DeepStream, leveraging its capabilities for efficient video processing and analysis.