TensorRT加速DBNet练习

最近在学习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是存放测试图片文件夹路径


未完待续...

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容