yolov5+tensorrt8 C++部署加速

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

注意–fp16

 

猜你喜欢

转载自blog.csdn.net/mao_hui_fei/article/details/129361860