pytorch,yolov5模型经onnx到Android(二)

以下为win10+protobuf+ncnn,主要从GitHub腾讯里面获取相关的步骤:

https://github.com/Tencent/ncnn

 

文章内容如下:

作为菜鸡,我还是将其全部列出来吧。方便后面再搞的时候,能快速上手。高手请略过。。。

1,安装vs2017社区版,这个不说。除了安装路径外,其他都可以选择下一个,安装路径如果C盘够,也可以直接默认。下载可去 i tell you

2,Start → Programs → Visual Studio 2017 → Visual Studio Tools → x64 Native Tools Command Prompt for VS 2017这句话就是如下图。点击开这个命令符工具

3,下载protobuf-3.4.0,解压。我将其解压到 D:\protobuf-3.4.0

cd D:\protobuf-3.4.0
mkdir build-vs2017
cd build
cmake -G"NMake Makefiles" -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=%cd%/install -Dprotobuf_BUILD_TESTS=OFF -Dprotobuf_MSVC_STATIC_RUNTIME=OFF ../cmake
nmake
nmake install

build文件我这边创建不了,所以直接命名build-vs2017

 4,下载安装vulkan sdk

5,下载ncnn,https://github.com/Tencent/ncnn 得到ncnn-master.zip。解压放到D:\ncnn-master

cd D:\ncnn-master
mkdir -p build-vs2017
cd build
cmake -G"NMake Makefiles" -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=%cd%/install -DProtobuf_INCLUDE_DIR=D:/protobuf-3.4.0/build-vs2017/install/include -DProtobuf_LIBRARIES=D:/protobuf-3.4.0/build-vs2017/install/lib/libprotobuf.lib -DProtobuf_PROTOC_EXECUTABLE=D:/protobuf-3.4.0/build-vs2017/install/bin/protoc.exe -DNCNN_VULKAN=OFF ..
nmake
nmake install

 注: -DNCNN_VULKAN=OFF设置成ON的时候不行。目前还没搞明白,先不管。跑完这一套,在D:\ncnn-master\build-vs2017\tools\onnx  下面会有onnx2ncnn.exe。

 

 到此,GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite 从这里开始得到的onnx权重文件就可以在上面进行拆分配置了

 将其复制到D:\ncnn-master\build-vs2017\tools\onnx,然后在进去D:\ncnn-master\build-vs2017\tools\onnx这个文件夹里面。

 

执行这条指令onnx2ncnn yolov5s.onnx yolov5s.param yolov5s.bin

得到相关文件,

 下面一篇就是如何配置param ,bin文件到Android里面去了

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

tiwolf_li

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值