使用ncnn调用pnnx转化的yolov5s6.pt,修改于ncnn源项目example中的yolov5_pnnx.cpp,在代码中修改引用的权值文件即可调用任何6.0版本权值。
需要在本系统中构建构建好ncnn后将ncnn/build/install中的include、lib复制到项目目录中的ncnn文件夹中,或在https://github.com/Tencent/ncnn/releases中下载好对应系统的预编译库放在项目目录中并且改写CMakeLists.txt中的ncnn目录。
需要opencv环境,作者构建于arch linux,贴出对应的linux命令:
sudo pacman -Syu hdf5 vtk opencv
由于linux最新版本的opencv目录为/usr/lib/opencv4/opencv2,建议将opencv2文件夹单独复制到/usr/lib中以兼容代码。
构建完环境后,只需要在项目文件夹里:
mkdir build & cd build
cmake ..
make
就可以在build文件夹里生成可运行的文件了,将对应的bin和param权值文件放到执行文件同目录下即可调用,调用样例为:
./yolov5v6_ncnn 0.jpg