[Caffe]:关于Check failed: shape[i] >= 0 (-2 vs. 0)错误

来源:互联网 发布:程序员要懂软件架构吗 编辑:程序博客网 时间:2024/06/05 10:48

错误发现及分析

当使用caffe训练网络的时候,出现如题错误,我们可以分析下日志,日志如下:
error
我们可以看出,在错误发生前生成的Top shape:64 2048 4 4 。也就是feature size是4 × 4的。然后再看看pool5对应的pooling layer的设置:
pool5
发现pooling的核大小被设置成7,这就是问题的所在了。

解决方案

  1. 改变输入图像的大小,可以使feature map size变大
  2. 更改kernel_size的大小
    第二种方法比较直接。
0 0
原创粉丝点击