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

文章目录

• 前言
• 一、配置tensorRT
• 二、使用vs2019 cuda10.2编译tensorrt代码
• • 1.首先把代码和模型文件下载备用
  • 2.配置sln/props里的配置文件
  • 3.配置sln/dll_detector里的dll_detector.vcxproj配置文件
  • 4.先build dll_detector这个项目
  • 5.build test_dll这个最后的项目
• 总结

---

前言

参考一个大佬的代码，知乎链接：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，我也是在他的指导下才弄好。
其他的功能慢慢去摸索源码了。。