模型实战（10）之win10下tensorRT部署yolov5算法

win10下tensorRT部署yolov5算法

• 针对yolov5的众多版本，目前有几种部署方法：

  • 1.tensorrtx:大神针对众多网络模型（包含yolov5众多版本）开源的一个转换工具，部署方式为：.onnx -> .wts -> .engine
  • 2.yolov5官方export.py导出onnx，通过C++或者python版的tensorrt函数接口进行编译（build），直接转换为.engine；或者通过tensorrt安装目录下的trtexec.exe对.onnx进行编译并转换为.engine

• 本文主要讲述如何通过tensorrtx 对 yolov5 进行部署

1.下载源码

• 下载源码+部署环境在此不再赘述了，如果安装虚拟环境之类的有啥问题可以参考我之前的文章，有详细的建立过程~
• yolov5-v6.0: https://github.com/ultralytics/yolov5/tree/v6.0