caffe进行finetune时出现"shapeequals(proto) shape mismatch (reshape not set)"的解决办法

声明加载的caffemodel会根据你的net.prototxt文件里的各个layer的name来进行参数赋值。

错误:[Caffe]: Check failed: ShapeEquals(proto) shape mismatch (reshape not set)

原因:从caffemodel加载进来的参数和你的输入或者输出不匹配。

示例:比如对于全连接层fc6,caffemodel中的bottom是1*256*6*6,而在你修改后的网络结构中,bottom是1*512*6*6,此时就会出现上述的错误。由于输入的shape是mismatch的。

解决:修改网络结构的同时,修改相应的layer的name,这样子在加载caffemodel的时候就不会加载pre-train的参数进来,从而避免参数和数据不匹配,但是这样做也不可避免的重新初始化了部分参数,不过fine-tune的目的刚好也在于此么。

原文地址:https://www.cnblogs.com/yyxf1413/p/7919310.html