当前位置:   article > 正文

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

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
推荐阅读
相关标签