TensorRT-YOLO: Support YOLOv5, YOLOv8, YOLOv9, PP-YOLOE using TensorRT acceleration with EfficientNMS!
Project description
English | 简体中文
🚀 TensorRT YOLO
TensorRT-YOLO 是一个支持 YOLOv5、YOLOv8、YOLOv9、PP-YOLOE 和 PP-YOLOE+ 的推理加速项目,使用 NVIDIA TensorRT 进行优化。项目不仅集成了 EfficientNMS TensorRT 插件以增强后处理效果,还使用了 CUDA 核函数来加速前处理过程。TensorRT-YOLO 提供了 C++ 和 Python 推理的支持,旨在提供快速而优化的目标检测解决方案。
✨ 主要特性
- 支持 YOLOv5、YOLOv8、YOLOv9、PP-YOLOE 和 PP-YOLOE+
- 支持 ONNX 静态、动态导出以及 TensorRT 推理
- 集成 EfficientNMS TensorRT 插件加速后处理
- 利用 CUDA 核函数加速前处理
- 支持 C++ 和 Python 推理
- CLI 快速导出与推理
🛠️ 环境要求
- 推荐 CUDA 版本 >= 11.6
- 推荐 TensorRT 版本 >= 8.6
📦 使用教程
📺 BiliBili
📄 许可证
TensorRT-YOLO采用 GPL-3.0许可证,这个OSI 批准的开源许可证非常适合学生和爱好者,可以推动开放的协作和知识分享。请查看LICENSE 文件以了解更多细节。
感谢您选择使用 TensorRT-YOLO,我们鼓励开放的协作和知识分享,同时也希望您遵守开源许可的相关规定。
📞 联系方式
对于 TensorRT-YOLO 的错误报告和功能请求,请访问 GitHub Issues!
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tensorrt_yolo-3.0.0.tar.gz
(19.5 MB
view hashes)
Built Distribution
Close
Hashes for tensorrt_yolo-3.0.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a7ed9ab828be3d2762a57e5138e8f986ca6c22c23d59a793d0d3a32c75548b99 |
|
MD5 | bdec9c53e41550bf1fc6d6a0e511129f |
|
BLAKE2b-256 | 8b74233af7b787089c125f361020b5b4763208db87159ee03948daa17bc12476 |