TrojanXu/yolov5-tensorrt
Fork: 46 Star: 190 (更新于 2024-11-02 15:15:44)
license: Apache-2.0
Language: Python .
A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5
最后发布版本: v3.0 ( 2020-10-04 15:51:07)
yolov5-tensorrt
A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5
requirement
Please use torch>=1.6.0 + onnx>=1.6.0 + TRT 7.1+ (fix upsample issue) to run the sample code
onnx-simplifier-0.2.16
The code
Add newly implemented upsample to get this working with current combination of onnx and tensorrt.
0. prepare above mentioned environment.
- git clone && git submodule update --init
- download weights file (use yolov5/models/export.py)
- python main.py to run the benchmark
- Generally, for image of size 640*640, using batchsize=1, the speedup is 4x on V100.
Updates
- 20201004 update to track yolov5 - v3.0 release. download model file from official websites please.
TODO
- NMS support
- dynamic shape or dynamic batchsize support (won't implement soon because onnx-simplifier only supports fixed shape)
- FP16 numerical issue and performance investigation
- Benchmark
最近版本更新:(数据更新于 2024-09-11 21:15:05)
2020-10-04 15:51:07 v3.0
TrojanXu/yolov5-tensorrt同语言 Python最近更新仓库
2024-11-06 03:34:16 home-assistant/core
2024-11-05 21:24:09 ai25395/FMatPix
2024-11-05 16:16:26 Guovin/TV
2024-11-05 15:03:24 Cinnamon/kotaemon
2024-11-04 23:11:11 DS4SD/docling
2024-11-04 10:56:18 open-compass/opencompass