1,安装环境
CUDA10.2
张量RT7.2
OpenCV3.4(工程中已指定,不需安装)
vs2015下载相关工程: :
2,生成yolov5s.wts文件
在生成yolov5s.wts前,首先需要模型。同时,需要我们安装环境。这里可以参考网上其他文章或github教程进行配置安装,这里不详加说明。
将tensorrtx-master\yolov5文件夹下的gen_wts.py拷贝到ultralytics/yolov5文件夹下
在当前目录下执行:
python gen_wts.py
最终我们会在当前目录下得到一个yolov5s.wts文件。
3,vs2015环境建设
这里我们使用别人已经编好的库,下载连接: 。(注意:部分头文件,lib文件已在我之后的工程中命名,有需要的同学可以直接下载)
创建vs工程,命名为yolov5_Trt,重新连接tensorrtx-master
2021-08-23 15:45:19
2.75MB
C++
1