模型实战(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
- 1.tensorrtx:大神针对众多网络模型(包含yolov5众多版本)开源的一个转换工具,部署方式为:
-
本文主要讲述如何通过tensorrtx 对 yolov5 进行部署
1.下载源码
- 下载源码+部署环境在此不再赘述了,如果安装虚拟环境之类的有啥问题可以参考我之前的文章,有详细的建立过程~
- yolov5-v6.0: https://github.com/ultralytics/yolov5/tree/v6.0