GitHub - arthurtibame/YOLOv5-TensorRT

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
yolov5		yolov5
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
main.py		main.py
upsample.py		upsample.py
yolo.py		yolo.py
yolov5s.yaml		yolov5s.yaml

Repository files navigation

YOLOv5-TensorRT

最近收到一块Xavier-NX的板子，想在上面部署一些常用模型试试性能，所以这是YOLOv5网络的TensorRT移植版本。

依赖

torch==1.4.0
onnx==1.6.0
TRT 7.0

代码说明

待补充。

About

No description, website, or topics provided.

Apache-2.0 license

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 99.0%
Other 1.0%