项目目的

目录介绍

mnist_cpp

9:30 mnist mnist python 训练代码
15:08 CMakeLists.txt cmake文件
21:39 CRnnDeploy.h
17:07 main.cpp main函数
16:47 run.sh 构建sh

yolov5_cpp

weights/last.torchscript.pt

face_detector_cpp(ncnn) 缺点: python和c++版差别有点大,尤其是在测试sample.jpg图片时,人脸检测的数量差别明显,有待优化

./run.sh

环境相关

1.Manjaro系统
2.gcc7
3.python3.6.10

环境搭建步骤

1.下载pytorch c++版:libtorch
2.安装opencv 可下载源文件自己构建也可直接pacman安装,请注意依赖问题,vtk包等

安装过程遇到的问题

1.在安装opencv时,无论是 sudo pacman -S opencv ,还是 下载二进制包编译安装,都是可以成功的, 测试也是没问题的
2.加入libtorch后开始报错
/usr/bin/ld: CMakeFiles/demo.dir/main.cpp.o: in function `main':
main.cpp:(.text+0x95): undefined reference to `cv::imread(std::string const&, int)'
原因肯定是哪里的版本不对,又重新下载了libtorch,不过这个会下的慢,解压后替换原来的,完美运行!
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/cu101/libtorch-cxx11-abi-shared-with-deps-1.6.0%2Bcu101.zip