caffe模型轉ncnn模型


下載MobileNet的caffe模型和配置文件

可從https://github.com/shicai/MobileNet-Caffe中下載,下載后得到mobilenet_deploy.prototxt和mobilenet.caffemodel兩個文件。


舊版caffe模型轉新版caffe模型

因為ncnn只支持轉換新版的caffe模型,所以需要先把第二步下載的caffe模型轉換為新版的caffe模型。新版caffe框架中自帶了轉換的工具,使用姿勢如下。

$ ~/caffe/build/tools/upgrade_net_proto_text mobilenet_deploy.prototxt mobilenet_deploy_new.prototxt
$ ~/caffe/build/tools/upgrade_net_proto_binary mobilenet.caffemodel mobilenet_new.caffemodel

 


新版caffe模型轉ncnn模型

在第一步生成的ncnn/build/tools目錄下用caffe2ncnn來轉換新版的mobileNet模型。

$./caffe2ncnn mobilenet_deploy_new.prototxt mobilenet_new.caffemodel mobilenet.param mobilenet.bin

  

 

來自 <https://blog.csdn.net/soralaro/article/details/81131615>


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM