caffe中 对caffemodel进行结构修改并保存

来源:互联网 发布:java开发课程进度计划 编辑:程序博客网 时间:2024/05/23 02:18

问题综述:利用matlab,在编译matcaffe的前提下,对已有的caffemodel进行结构修改,例如,删除一些层等


问题描述:在深度学习过程中,自己的目标任务往往需要基于物体识别等等其他任务的基础上,加上自己的训练数据不足,这就需要我们对已有的网络进行拷贝权重,用自己的数据进行fine-tune,从而达到自己的目的


解决方法:

首先需要matcaffe,

运行以下代码即可

caffe.reset_all();##启动caffenet = caffe.Net('after-modify-deploy.prototxt', 'original.caffemodel', 'test');##根据修改的
##deploy文件对caffemodel进行修改 net.save('after-modify.caffemodel');##将修改后的caffemodel进行保存
所需要的修改 皆在原有model的deploy文件中进行修改,

而后执行该代码即可

原创粉丝点击