使用caffe的python接口裁剪caffemodel

2 篇文章 0 订阅
1 篇文章 0 订阅

最近需要将训练好的caffemodel裁剪保留前几层,然乎保存为新的caffemodel

找了一些资料都是解析或修改caffemodel参数,和结构相关的只有把caffemodel的最后的fc层通过flat转化成卷积层。

找到一篇相关的,只不过是用caffe的matlab接口实现的。由于不太好找,所以在这里记录下链接:https://blog.csdn.net/wangkun1340378/article/details/76474900

验证了下,python和matlab代码是一样的。

具体方法:copy已有的deploy文件保存为deploy_copy2.prototxt文件,然后修改,删掉后面不需要的层,然后保存
接着输入以下代码:先加载模型,然后保存新模型即可。
在这里插入图片描述
结果:
在这里插入图片描述
后续验证查看caffemodel的结构和参数可以用netron工具。这个工具既可以通过训练好的caffemodel来查看网络结构和参数,也可以通过模型设置文件(如deploy.prototxt)来查看网络结构。
具体方式可以通过左上角的file打开对应文件,也可以直接把文件拖入此窗口自动打开。
新的模型结构:
在这里插入图片描述
原来的模型结构:
在这里插入图片描述

另外查看具体参数可以点击模块,右侧出现的信息就是该层的参数:
在这里插入图片描述
点击INPUTS里的filter和bias右侧的"+"号可以看到参数的值:
在这里插入图片描述

附:关于caffemodel参数的解析:
Caffe学习:使用pycaffe读取caffemodel参数 - from_jiarenyf - CSDN博客 https://blog.csdn.net/u011762313/article/details/49851795

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值