gm0515
gm0515
采纳率50%
2018-10-30 03:34

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条回答

  • caozhy 回答这么多问题就耍赖把我的积分一笔勾销了 3年前

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

    点赞 评论 复制链接分享
  • gm0515 gm0515 3年前

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

    点赞 评论 复制链接分享