devbox
羊村懒王
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

Vs21019 tensorRT 加速yolov5 生成engine推理模型（C++）_wts转engine

作者：羊村懒王 | 2024-04-14 20:03:19

赞

踩

wts转engine

所需软件/组件

Vs2019

【精选】visual studio 2019安装教程（详细的很）_vs2019安装教程-CSDN博客

opencv

如何在VS2019中配置OpenCV_vs2019 opencv-CSDN博客

Cuda

【精选】CUDA安装教程（超详细）_Billie使劲学的博客-CSDN博客

(注意！这个连接教程中并未勾选)

一定要勾选，否则报错找不到cuda

Cudnn (上面的连接里包含了cudnn安装，很详细)

CMake

CMake的安装（超级详细）_cmake安装_小码1111的博客-CSDN博客

Yolo源码

https://raw.gitcode.com/mirrors/monday-leo/yolov5_tensorrt_win10/archive/refs/heads/master.zipicon-default.png?t=N7T8https://raw.gitcode.com/mirrors/monday-leo/yolov5_tensorrt_win10/archive/refs/heads/master.zip

TensorRT

【模型部署】TensorRT的安装与使用_tensorrt部署-CSDN博客

CMake编译模型

（将yolov5的模型放在一个方便找到的文件夹目录下后面要用）

  1. 在yolov5的源文件夹下手动添加一个名为bulid的文件（Yolov5_Tensorrt_Win10-master\bulid）

  2. 修改yolov5源码文件中的CMakeLists.txt文件

（如何查看自己计算机算力）

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/羊村懒王/article/detail/423869
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。