Yolov5 win10平台下的C++推理及Tensorrt部署 篇一 (共三篇)

本文介绍了在Windows 10环境下,使用C++进行Yolov5 v4.0的TensorRT部署。由于TensorRT 5.0-7.0仅支持Linux,故Yolov5 v5.0和6.0无法直接部署。文章详细阐述了从转换权重到C++调用的步骤,包括训练模型、权重转换、下载C++推理文件、安装OpenCV库,以及确保CUDA和cuDNN版本匹配。通过TensorRT,可以实现模型的加速推理,提高实时性并减少显存占用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

为什么用Yolov5而不用其他的网络:

  • 目标检测项目落地一般都是用Yolov5和YoloR来做。其一是因为精度速度确实比较牛逼,其二是因为这两个网络相对于其他的先进网络有较为成熟的C++推理方案。
  • 因为要在Win10下部署,Linux的一些方案就不能拿来用。比如YOLOX尽管可以用C++推理,但将其pth权重转为TensorRT的engine权重却会存在问题,因为TensorRT的5.0-7.0版本只支持Linux系统。

温馨提示:我们的部署版本是Yolov5 - 4.0,Yolov5 - 5.0或6.0亲测不支持。也就是说,如果用其他版本的Yolov5训练出来的网络,目前是不支持用C++调用的。同时,C++部署的环境比较苛刻,不同平台的权重不能通用,甚至显卡不一样也会报错,不过按照以下流程一步一步来,就不会有问题啦。关于如何在win10下配置Yolov5 - 4.0的训练环境,请点击这里:

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

刚哥吧

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值