/YOLOv5-TensorRT

Primary LanguagePythonApache License 2.0Apache-2.0

YOLOv5-TensorRT

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

依赖

  • torch==1.4.0
  • onnx==1.6.0
  • TRT 7.0

代码说明

待补充。