声明加载的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的目的刚好也在于此么。

posted on 2017-11-29 11:17  2师兄不会胖  阅读(519)  评论(0编辑  收藏  举报