yolov5+tensorrt C++部署加速
参考代码:
主要参考github:https://github.com/yzy12-max/yolov5_deploy
码云备份:https://gitee.com/qitiandashengdiyun/yolov5_tensorrt8
https://github.com/FeiYull/TensorRT-Alpha
yolov5 pt模型转onnx
先在官网下载yolov5最新版代码和最新版模型(截止2023年3月6日,是yolov7-7.0)
使用命令
python export.py --weights yolov5l.pt --include onnx engine --img 640
onnx转engine
使用官方自带的工具
不同的机器位置可能有所
cd /usr/src/tensorrt/bin
trtexec -
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)