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

本文分析了在使用Caffe进行网络训练时遇到的一个错误,该错误与特征图尺寸和池化层设置有关。通过调整输入图像大小或更改核大小可以解决此问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

错误发现及分析

当使用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的大小
    第二种方法比较直接。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值