使用训练好的caffemodel进行微调

版权声明:学无止境,好好学习 https://blog.csdn.net/m0_38116269/article/details/88947232

加油,之前简单尝试过,直接改会遇到caffemodel里面的1000个参数出和自己需要做的10分类输出不匹配的问题。希望这次可以解决。

参考1:https://www.cnblogs.com/louyihang-loves-baiyan/p/5038758.html
貌似写的很详细,可以借鉴,这一篇写的真是太好了,成功解决了我遇到的问题。大赞!

参考2:https://www.cnblogs.com/denny402/p/5137534.html

竟然可以了,成功了,关键点:
要想导入预训练的模型来做微调的话,一定要记得不仅要修改全连接层的num_output,更重要的是要修改这一层的名字,和top层的名字,因为求解器正是通过查名字,如果名字不一样了,我就得把caffemodel里面的权重重新来过来适应你正在训练的模型,不然就报错无法训练啦。我上次距离解决这个bug,就只差这一个小tip啊。

猜你喜欢

转载自blog.csdn.net/m0_38116269/article/details/88947232