最近在学习TensorRT过程中。github上发现DBNet项目后,就以此项目进行练习。
github项目见链接:https://github.com/BaofengZan/DBNet-TensorRT
题记
原本非常丝滑的练习过程,以为半个小时能搞定,结果搞了一天。
主要问题是按原作者步骤进行cmake生成工程时无法生成工程,提示如下问题:
关于这个问题,网上有说是本地电脑中cuda(10.0,10.1)存在多个版本导致。一顿操作猛如虎(百度不到解决办法)后,果断放弃cmake方法(菜是原罪),自己新建工程配置TensorRT及OpenCV环境。
平台说明:win10+vs2015 update3+TensorRT7.0+OpenCV4.4.0
主要步骤如下:
1. 配置环境
参考链接:https://www.jianshu.com/p/120897d69dca
参考链接中步骤配置好TensorRT环境;百度一篇vs2015+OpenCV4.4.0配置教程,按步骤配置即可(TensorRT+OpenCV可同时配置)
注意:在配置之前搞清楚TensorRT是Release版本还是Debug版本(小声告诉你我配了两遍)。
2. 添加工程文件
将原作者工程中的代码文件添加进新建工程
3. 生成解决方案,生成.exe文件
4. 生成engine(请忽略狂放的马赛克)
控制面板运行.exe -s
5. 测试并查看检测效果
控制面板运行.exe -d ./folder
注:folder是存放测试图片文件夹路径
未完待续...