caffe转ncnn

最新推荐文章于 2021-12-24 15:12:31 发布

上天夭

最新推荐文章于 2021-12-24 15:12:31 发布

阅读量542

点赞数

CC 4.0 BY-SA版权

分类专栏： Caffe

本文链接：https://blog.youkuaiyun.com/weixin_38952721/article/details/94747999

Caffe 专栏收录该内容

3 篇文章

订阅专栏

本文详述从旧版Caffe模型升级至新版，并进一步转换为NCNN模型的全过程。涉及模型文件更新、参数调整及转换命令，适用于希望将预训练模型部署于移动端或嵌入式设备的开发者。

caffe转ncnn

convert文件夹下，转换之前，存在 bvlc_alexnet.caffemodel 和 deploy.prototxt两个文件，转换之后，新增加了 alexnet.bin ， alexnet.param ， new_bvlc_alexnet.caffemodel ，new_deploy.prototxt

1、已知文件

~/caffe/lhh/convert$ ls
bvlc_alexnet.caffemodel  deploy.prototxt

2、旧版caffe模型转新版caffe模型

~/caffe/lhh/convert$ ~/caffe/build/tools/upgrade_net_proto_text deploy.prototxt new_deploy.prototxt

~/caffe/lhh/convert$  ~/caffe/build/tools/upgrade_net_proto_binary bvlc_alexnet.caffemodel new_bvlc_alexnet.caffemodel

注意：upgrade_net_proto_text 和 upgrade_net_proto_binary要修改成自己对应的路径

3、new_deploy.prototxt的修改

因为我们每次检测一张图片,所以要对新生成的deploy.prototxt进行修改:第一个 dim 设为 1

~/caffe/lhh/convert$ vi new_deplpy.prototxt

在这里插入图片描述

4、新版caffe模型转ncnn模型

~/caffe/lhh/convert$ ~/liq/ncnn/build/tools/caffe/caffe2ncnn new_deplpy.prototxt new_bvlc_alexnet.caffemodel alexnet.param alexnet.bin

转换成功

~/caffe/lhh/convert$ ls
alexnet.bin  alexnet.param  bvlc_alexnet.caffemodel  deploy.prototxt  new_bvlc_alexnet.caffemodel  new_deplpy.prototxt

5、总结：简而言之，一共三行代码

~/caffe/build/tools/upgrade_net_proto_text deploy.prototxt new_deplpy.prototxt

~/caffe/build/tools/upgrade_net_proto_binary bvlc_alexnet.caffemodel new_bvlc_alexnet.caffemodel

~/liq/ncnn/build/tools/caffe/caffe2ncnn new_deploy.prototxt new_bvlc_alexnet.caffemodel alexnet.param alexnet.bin