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

本文介绍了如何在Linux系统上配置并编译NCNN库,包括必要的依赖安装,以及使用`onnx2ncnn`工具将ONNX模型转换为NCNN格式的过程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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

### 如何在 Ubuntu 20.04安装配置 YOLOv5 #### 安装依赖项 为了确保YOLOv5能够正常运行,在Ubuntu 20.04上需先安装必要的软件包。这包括更新现有的包列表并安装Python3-pip和其他一些工具。 ```bash sudo apt-get update && sudo apt-get upgrade -y sudo apt install python3-pip git wget -y ``` #### 设置 Python 环境 创建一个新的虚拟环境来隔离项目所需的库文件,防止与其他项目的冲突。 ```bash pip3 install virtualenv virtualenv yolo_env source yolo_env/bin/activate ``` #### 下载 YOLOv5 源码 克隆官方GitHub仓库到本地机器以便获取最新的源代码版本。 ```bash git clone https://github.com/ultralytics/yolov5.git cd yolov5 ``` #### 安装所需库 通过`requirements.txt`文件自动下载并安装所有必需的Python模块。 ```bash pip install -r requirements.txt ``` #### 配置 CUDA 和 cuDNN 对于GPU加速的支持至关重要,因此需要正确设置NVIDIA驱动程序以及相应的CUDA Toolkit和cuDNN SDK版本[^2]。 - 参考 NVIDIA 官方文档选择合适的驱动版本。 - 使用 `apt` 或者.run 文件方式完成CUDA toolkit 的部署。 - 将 cuDNN 库复制至相应目录下,并调整环境变量使其生效。 #### 测试安装情况 最后一步是验证整个流程是否成功: ```python import torch from models.experimental import attempt_load model = attempt_load('yolov5s.pt', map_location=torch.device('cpu')) # or 'cuda' if GPU is available print(model) ``` 如果一切顺利,则会显示模型结构信息;反之则可能遇到错误提示,此时应仔细排查每一步骤中的潜在问题。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值