win10 下 tensorRT 部署 YOLOV5模型详解

tensorRT 部署 YOLOV5模型详解
第一步:
下载tensorRT库
https://developer.nvidia.com/nvidia-tensorrt-8x-download
解压后,编译

在这里插入图片描述
第二步:cmakeList.txt需要进行修改,·改成自己的路径
在这里插入图片描述
第三步:下载dirent-master 将dirent.h文件copy到src文件夹下
在这里插入图片描述

第四步:使用的opencv_world460.dll 拷贝进环境变量
在这里插入图片描述
第五步 参数进行修改
在这里插入图片描述
第六步
6.1、pt->wts
python gen_wts.py -w yolov5s.pt -o yolov5s.wts
6.2、wts -----> engine
yolov5_det.exe -s yolov5l_best.wts yolov5l_best.engine l
在这里插入图片描述
第七步 保持yolov5与tensorrt版本一致
在这里插入图片描述
第八步、cmake即可,打开生成工程
在这里插入图片描述
总结:过程中需要注意两点:a、主要配置自己的参数;b、cmakeLists文件中的路径修改;

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值