在运行caffe的过程中出现过几次这个错误,当时都是解决了就每管它了,今天再出现,决定记录下可能原因以及解决方法:
检查定位到blob中的455,发现可能是 N C H W中某个出错.
1 在做finetune时,需要确保网络的crop一致,比如,比如好多用ImageNet的网络裁剪尺寸就不同,比如AlexNet中是227,而到了 VGGNet中则变成了224,这样的细微差别只能细心查验了.
2 利用训练好的model来做predict,
input_dim: 10
input_dim: 3
input_dim: 64
input_dim: 64 要注意原始的deploy中,预测图片张数为多少
3 如果不是以上原因,请尝试:在net.cpp中的742行设置kReshape = true
本文参考链接:https://github.com/BVLC/caffe/issues/2834
https://groups.google.com/forum/#!topic/caffe-users/riTWI6nlIJ4
本文针对Caffe训练过程中出现的shape不匹配错误进行了解析,并提供了详细的排查步骤及解决方案,包括确保网络输入尺寸一致、调整预测图片数量以及修改源代码等方法。
1081

被折叠的 条评论
为什么被折叠?



