下载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>
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:caffe模型转ncnn模型 - Python技术站