前言
本文是视频https://www.bilibili.com/video/BV1dp4y177L4的笔记。
OpenCV和Libtorch安装包:https://pan.baidu.com/s/1i3DqTcHFSC1rRDsIgYGCsQ?pwd=8888
VS版本:2019
Opencv版本:3.4.1
Libtorch版本:2.0.1+cu117
配置OpenCV环境
1.打开VS,创建控制台应用。
2.在视图中,调出属性管理器。
3.属性管理器中,选择Debug|x64的属性。
4.在包含目录中添加OpenCV的路径,我这里是E:\C_Libiary\opencv3.41\build\include
和E:\C_Libiary\opencv3.41\build\include\opencv2
5.库目录添加E:\C_Libiary\opencv3.41\build\x64\vc15\lib
6.附加依赖项添加opencv_world341d.lib
7.环境变量添加E:\C_Libiary\opencv3.41\build\x64\vc15\bin
8.将bin目录下的三个dll文件复制到C:\Windows\System32
路径中
9,在cpp文件下放置一张图片,调用下面代码进行测试
#include <iostream>
#include <opencv2/opencv.hpp>
using namespace cv;
using namespace std;
int main()
{
Mat src = imread("1.png"); /*图片 */
imshow("input", src);
waitKey(0);
return 0;
}
成功打开,说明OpenCV环境配置成功。
配置Libtorch环境
1.在包含目录中添加Libtorch的路径,我这里是E:\C_Libiary\libtorch\include\torch\csrc\api\include
和E:\C_Libiary\libtorch\include
2.添加库目录,路径为E:\C_Libiary\libtorch\lib
3.在依赖项添加以下内容(Libtorch文件夹下的一些lib文件,不同版本的Libtorch会略有区别)
asmjit.lib
c10.lib
c10_cuda.lib
caffe2_nvrtc.lib
clog.lib
cpuinfo.lib
dnnl.lib
fbgemm.lib
fbjni.lib
kineto.lib
libprotobuf.lib
libprotobuf-lite.lib
libprotoc.lib
nvfuser_codegen.lib
pthreadpool.lib
pytorch_jni.lib
torch.lib
torch_cpu.lib
torch_cuda.lib
XNNPACK.lib
4,添加环境PATH=E:\C_Libiary\libtorch\lib;%PATH%
5.复制所有的dll文件到C:\Windows\System32
路径中
6.输入以下代码进行测试
#include<torch/torch.h>
#include<torch/script.h>
using namespace torch;
using namespace std;
int main() {
Tensor tensor = torch::rand({ 1,2,3 });
cout << tensor.sizes() << endl; //方式一,只打印维度信息
tensor.print(); //方式二,除了打印维度信息,数据类型也打印出来
return 0;
}
成功运行,则表示配置成功。