🚀 TensorRT YOLO
TensorRT-YOLO is an inference acceleration project that supports YOLOv3, YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv9, YOLOv10, YOLO11, PP-YOLOE and PP-YOLOE+ using NVIDIA TensorRT for optimization. The project not only integrates the TensorRT plugin to enhance post-processing effects but also utilizes CUDA kernel functions and CUDA graphs to accelerate inference. TensorRT-YOLO provides support for both C++ and Python inference, aiming to deliver a fast and optimized object detection solution.
✨ Key Features
- Support for YOLOv3, YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv9, YOLOv10, YOLO11, PP-YOLOE and PP-YOLOE+
- Support Detect and OBB Detect models
- Support for ONNX static and dynamic export, as well as TensorRT inference
- Integration of TensorRT plugin for accelerated post-processing
- Utilization of CUDA kernel functions for accelerated preprocessing
- Utilization of CUDA graphs for accelerated inference process
- Support for inference in both C++ and Python
- Command-line interface for quick export and inference
- One-click Docker deployment
🛠️ Requirements
- Recommended CUDA version >= 11.6
- Recommended TensorRT version >= 8.6
📦 Usage Guide
📺 BiliBili
☕ Buy the Author a Coffee
Open source projects require effort. If this project has been helpful to you, consider buying the author a coffee. Your support is the greatest motivation for the developer to keep maintaining the project!
It's recommended to use Alipay, as WeChat doesn't provide the avatar. Please note "TensorRT-YOLO" in the transfer.
📄 License
TensorRT-YOLO is licensed under the GPL-3.0 License, an OSI-approved open-source license that is ideal for students and enthusiasts, fostering open collaboration and knowledge sharing. Please refer to the LICENSE file for more details.
Thank you for choosing TensorRT-YOLO; we encourage open collaboration and knowledge sharing, and we hope you comply with the relevant provisions of the open-source license.
📞 Contact
For bug reports and feature requests regarding TensorRT-YOLO, please visit GitHub Issues!
1 maintainer
Metadata
1 maintainer
Release files for tensorrt-yolo 4.3.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| tensorrt_yolo-4.3.1.tar.gz | 67.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| tensorrt_yolo-4.3.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 130.7 kB
Release files / tensorrt_yolo-4.3.1.tar.gz
| Download URL | tensorrt_yolo-4.3.1.tar.gz |
|---|---|
| Size | 67.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
cad21c2da647d29736b4032ce91ceec7747dabf3945aed51cb0e7dea3b7de812
|
|
BLAKE2b-256 checksum How to use checksums |
f4ba6c897ac0a1c610501151095d11cb987e53b71888fb7e3fe1bd85bacfbb5a
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.12.3
|
Release files / tensorrt_yolo-4.3.1-py3-none-any.whl
| Download URL | tensorrt_yolo-4.3.1-py3-none-any.whl |
|---|---|
| Size | 62.8 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
90dca78fc06eabea43f879c0fef169d8c18139be0bab11220300208471106baf
|
|
BLAKE2b-256 checksum How to use checksums |
19e989131a22c60c033ec72b2fd4974f16bf2d8e8e2f8c8fd2e53db65aa3acc3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.12.3
|
