Caffe -SSD 训练报错,Cannot copy param 0 weights from layer 'conv1_1'; shape mismatch.

F1030 11:27:00.626530 31870 net.cpp:774] Cannot copy param 0 weights from layer 'conv1_1'; shape mismatch. Source param shape is 64 3 3 3 (1728); target param shape is 64 4 3 3 (2304). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

没有积分,如能解决可现金答谢!

2个回答

目测conv1_1的维度搞错了,一个是64 3 3 3,一个是64 4 3 3,要看你的代码,把多余的数据维度用切片切掉,或者把模型中conv1_1的维度补齐

gm0515
gm0515 谢谢 ,找到原因了。
一年多之前 回复

找到原因了,输入图像的通道是4通道,png格式的问题,第一卷积层的输入应为3通道。

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问
相关内容推荐