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