python caffe 识别文件出现参数尺寸错误

错误类型

Memory required for data: 432688
F0601 14:55:27.226405 329656 net.cpp:141] Check failed: param_size <= num_param_blobs (1 vs. 0) Too many params specified for layer drop4
*** Check failure stack trace: ***
Aborted (core dumped)

得到caffe模型后,采用模型进行测试识别一张图片,出现参数错误, 错误出现在 fc4 层, 排查后,发现在 deploy.prototxt 文件中, Input 层的数据shape 尺寸设置过小,经过层层 conv -> pool -> fc 时,特征数据尺寸不足,修改输入尺寸后成功运行。

原输入层

layer {
  name: "data"
  type: "Input"
  top: "data"
  input_param{shape:{dim: 1 dim: 3 dim: 32 dim:32}}
}

修改后的输入层

layer {
  name: "data"
  type: "Input"
  top: "data"
  input_param{shape:{dim: 1 dim: 3 dim: 48 dim:48}}
}

版权声明:本文为qq_21476007原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。