zl程序教程

您现在的位置是:首页 >  .Net

当前栏目

TensorRT加速(VS+opencv4.5)

2023-02-18 16:27:16 时间

1.安装配置TensorRT

(1)在官网进行下载:https://developer.nvidia.cn/zh-cn/tensorrt

 

 (2)配置

包含目录:

 

 库目录:

 

 链接器:(3个lib路径中的.lib文件都需要)

 Tip:

可以在路径中输入cmd,弹出命令窗口

然后在该路径下输入 dir ./b 回车

 

 

 2.创建项目

Tips:

  • 如果建立的是c++空项目,有可能会报错“找不到cublas.lib"
  • 如果找不到NVIDIA,创建c++项目后,选择项目右键——生成依赖项——生成自定义——(里面没有CUDA的话)查找现有的——路径如下图

 

 

 3. 代码

参考:https://cloud.tencent.com/developer/article/1800743

https://blog.csdn.net/hjxu2016/article/details/119796206

Tips:

1. 先创建内存空间,再加载图进行推理

2. 图片预处理

resize(image, image, Size(224, 224));
image.convertTo(img2, CV_32F);
img2 = (img2 / 255 - 0.2458) / 0.0612;

相当于:

transforms.Resize([224, 224]),
transforms.ToTensor(),
transforms.Normalize(mean=[0.2458], std=[0.0612])

3. 代码整理好了再放