Cannot copy param 0 weights from layer 'conv1'; shape mismatch.

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/kele_imon/article/details/79860785

当我使用训练好的caffemodel准备测试一下图片的时候,我定义了deploy.prototxt文件,然后利用bat脚本进行测试,发现报错,错误如下:


错误解决:打开我的deploy.prototxt文件,把Input_param中的数据改掉,把1通道改为3通道,shape参数的意思:第一个参数表示batch_size,第二个参数表示通道(彩色为3通道),剩下两个表示图像大小,我用的是黑白图像,所以改成了1通道,但是不行,所以我就又改了回去,再次运行bat脚本,就发现可以使用了。

input_param修改如下:

至于deploy.prototxt文件如何定义,这时建立在网络文件上的,修改方法详见此文章:https://blog.csdn.net/kele_imon/article/details/79106942

祝大家在caffe的路上一路顺利。

猜你喜欢

转载自blog.csdn.net/kele_imon/article/details/79860785