ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

win10 vs2019 tensorRT7 yolov5配置

2021-01-29 14:02:34  阅读:1155  来源: 互联网

标签:yolov5 配置文件 vs2019 代码 tensorrt dll tensorRT7 win10 detector


提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

文章目录


前言

参考一个大佬的代码,知乎链接:https://zhuanlan.zhihu.com/p/102453331
GitHub代码链接:https://github.com/enazoe/yolo-tensorrt

一、配置tensorRT

参考之前的博客文章或者网上其他教程。

二、使用vs2019 cuda10.2编译tensorrt代码

1.首先把代码和模型文件下载备用

2.配置sln/props里的配置文件

在这里插入图片描述
把里面cuda opencv tensorrt这三个包的路径修改为自己的路径

3.配置sln/dll_detector里的dll_detector.vcxproj配置文件

<Import Project="$(VCTargetsPath)\BuildCustomizations\CUDA 10.2.props" />

因为原始是11.0我使用的是10.2所以要修改像上面那样(有两个地方)

如果报错没有props文件:
就把下面几个cuda文件的路径复制到c盘vs那个报错的路径里。
在这里插入图片描述

4.先build dll_detector这个项目

修改项目属性,因为vs2019有个api改了所以改成用2015build就能用。
在这里插入图片描述

5.build test_dll这个最后的项目

这个没改成2015也可以编译出来,要依赖上一个项目生成的detector.lib就是了

然后把模型放到…/configs/yolov5-3.0/目录下
把sample_detector.cpp 里改成yolov5版本或者其他支持的版本

detector->init(config_v5);

总结

非常的厉害的一个大佬写的代码,膜拜ing,我也是在他的指导下才弄好。
其他的功能慢慢去摸索源码了。。

标签:yolov5,配置文件,vs2019,代码,tensorrt,dll,tensorRT7,win10,detector
来源: https://blog.csdn.net/wenghd22/article/details/113375716

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有