Ubuntu20.04配置NCNN推理框架（转换yolov5 onnx格式到ncnn格式

想写C++的AI初学者

已于 2023-09-07 10:27:15 修改

阅读量823

点赞数 1

文章标签： YOLO 深度学习 ubuntu

于 2023-09-07 10:18:46 首次发布

本文链接：https://blog.csdn.net/weixin_51639169/article/details/132732078

版权

1.配置NCNN库必要依赖

sudo apt install build-essential git cmake libprotobuf-dev protobuf-compiler libvulkan-dev vulkan-utils libopencv-dev

2.编译NCNN

执行下列命令

git clone https://github.com/Tencent/ncnn.git

cd ncnn

git submodule update --init

mkdir build

cd build

cmake ..

make -j8

make install

编译完成进入下列文件夹 ncnn/build/install/bin 在这里可以实现转换模型 onnx2ncnn

例如我的如下

./onnx2ncnn是命令，后面跟模型的位置

./onnx2ncnn /media/xcy/dcd05f09-46df-4879-bfeb-3bab03a6cc3a/YOLOv5-Lite/weights/yolov5-lite-sim.onnx /media/xcy/dcd05f09-46df-4879-bfeb-3bab03a6cc3a/YOLOv5-Lite/weights/yolov5-lite.param /media/xcy/dcd05f09-46df-4879-bfeb-3bab03a6cc3a/YOLOv5-Lite/weights/yolov5-lite.bin