win10 vs2019 tensorRT7 yolov5配置
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
文章目录
- 前言
- 一、配置tensorRT
- 二、使用vs2019 cuda10.2编译tensorrt代码
- 1.首先把代码和模型文件下载备用
- 2.配置sln/props里的配置文件
- 3.配置sln/dll_detector里的dll_detector.vcxproj配置文件
- 4.先build dll_detector这个项目
- 5.build test_dll这个最后的项目
- 三、int8 和 fp16 和 batch inference功能使用
- 1、int8
- 2、fp16
- 3、batch inference
- 总结
前言
参考一个大佬的代码,知乎链接: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文件:如下
error : 找不到导入的项目“C:\Program Files (x86)\MSBuild\Microsoft.Cpp\v4.0\V140\BuildCustomizations\CUDA 10.2.targets”。
请确认 Import 声明“C:\Program Files %28x86%29\MSBuild\Microsoft.Cpp\v4.0\V140\\BuildCustomizations\CUDA 10.2.targets”中的表达式正确,且文件位于磁盘上。
就把下面几个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);
三、int8 和 fp16 和 batch inference功能使用
1、int8
我使用的是cuda10.2,最新的tensorrt最好使用cuda11,我降版本使用tensorRT7.0后解决了问题,
同时calibration_images.txt里面改为测试图片的路径。
目前tensorRT7.0 cuda10.2不支持int8的批推理,单张可以。不知道最新版本修复没有。
2、fp16
同样是注意cuda的版本和tensorRT的版本,同时fp16支持批推理功能。
3、batch inference
需要使用scripts的yaml2cfg.py的转换,修改–batch 导出的模型才支持批推理,注意使用的是yolov5的v3.0的版本。
总结
非常的厉害的一个大佬写的代码,膜拜ing,我也是在他的指导下才弄好。
其他的功能慢慢去摸索源码了。。
待添加。。
win10 vs2019 tensorRT7 yolov5配置相关推荐
- 【win10 VS2019 opencv4.6 配置参考】
目录 1.下载 2. 添加环境变量 解压 添加环境变量 3.VS2019配置 第一种 第二种 opencv的VS环境配置不过三分钟的事情,但网上五花八门的配置看的眼花缭乱,不知道该说opencv个版本 ...
- 【chromium】win10 VS2019 环境 chromium 配置与编译。
文章目录 硬件&软件配置 安装 depot_tools get the code 编译 编译加速 基础概念 https://chromium.googlesource.com/chromium ...
- opencv2.4.13在win10+VS2015下的配置过程
opencv2.4.13在win10+VS2015下的配置过程 参考https://blog.csdn.net/shanghairuoxiao/article/details/69391229一个是本 ...
- cppunit在vs2019上的配置和使用【Win10】【详细】
CPPUnit在VS2019上的配置和使用[win10] tags: CPPUnit,软件测试,C++ 参考文章:https://www.cnblogs.com/xiaodi914/p/5251397 ...
- win10+vs2019+FFTW64位安装配置保姆教程
win10+vs2019+FFTW64位安装配置保姆教程 FFTW 是一个C语言的快速傅立叶变换库.(据说是世界上最快的FFT哦) vs2019下载与安装 参考链接:VS2019安装教程 FFTW下载 ...
- win10+vs2019配置lite.ai.toolkit预编译库
目录 一.配置OpenCV 2. 解压安装包,双击安装包,指定解压缩目录 3.属性页会在后面一起配置 二.配置Onnxruntime 三.配置TNN 四.在VS2019下编译MNN 五.配置NCNN环 ...
- 【opencv安装和配置完整版教程】(win10+vs2019+opencv4.4.0+opencv_contrib-4.4.0+永久配置)
[opencv安装和配置](win10+vs2019+opencv4.4.0+opencv_contrib-4.4.0+永久配置) 下载vs2019 opencv4.4.0.opencv_contri ...
- Win10+vs2019配置与运行RenderMatch+用contextcapture进行重建
Win10+vs2019配置与运行RenderMatch+用contextcapture进行重建 继上一篇博客,ground image 和aerial image影像匹配结果将会被保存为match. ...
- 【SLAM】SLAM环境配置 Win10+VS2019+OpenCV+PCL+g2o+Vcpkg
目录 SLAM环境配置 Win10+VS2019+OpenCV+PCL+g2o+vcpkg 软件及版本 1. Windows 10 2. Visual Studio 2019 3. OpenCV 4. ...
最新文章
- Handler消息机制(八):Handler内存泄漏的场景有哪些,如何避免
- MD5加密解密帮助类
- python基础知识选择题-python基础知识练习题(二)
- PHP ERROR_php中的异常和错误浅析
- java两字符串是否相等_Java与JavaScript中判断两字符串是否相等的区别
- stl reserve_vector :: reserve()函数以及C ++ STL中的示例
- MySQL面试题中:主从同步的原理
- 关于ASP.NET与CLR相互关系的一些总结
- 排序算法之三 选择排序(C++版本)
- 编译原理实验二 语法分析程序设计与实现
- Python 实现一个自动下载小说的简易爬虫
- 【图文教程】Shell基础知识
- SSD固态盘 _测试总结
- 位图(BMP)文件格式(一)
- 改善webERP运行速度很慢的问题
- linux 系统时间 硬件时间 及 时区设置
- 51单片机农历转换公历c语言算法,C51写的51单片机用公历转农历和星期程序
- 关于“24点”游戏的解法
- 【已解决】连接被重置
- JAVA根据经纬度计算两点距离