向上取整的方法_通过修改源码解决caffe的pooling向上取整的问题

caffe框架下载入模型的时候,经常会遇到这样的bug。

Check failed: bottom[0] ->shape() == bottom[i] ->shape() bottom[0]: 1 64 56 56 (200704), bottom[1]: 1 64 57 57 (207936)。

8f87fdaf8a2a46fd17cc37abd1820c2f.png

后来通过比对prototxt的layer标号和这里的错误信息,才知道是pooling层的锅。pooling层默认是向上取整的,或者更严谨地说应该是:【先加1再向下取整】。我在这一层本来输入的shape是1 64 112 112,结果经过它的运算变成了1 64 57 57。所以会出现和另一个分支shape不匹配的情况。

修改方法:在caffe的源码中,打开caffe/blob/master/src/caffe/ayers/pooling_layer.cpp

把这个地方的+1删除。直接在文本中搜索"+ 1"就可以找到这个位置。

eb685a44dc91d035b469ae5faa325284.png

3aa04aa14ebfba4f46ecfc9c5f94fcef.png

代码修改过后,如果你的caffe以前是编译过的,那么重新编译代码即可使用。

make clean

make pycaffe

make all

make test

然而,这个问题并没有因此解决。

68cc6eda73502ddac93ba60ffd2d3880.png

pooling的结果从57变成了55.

后来,我就直接用了莽夫操作:

4de229c33b665980f2de4adb627d0c26.png

莽夫一时爽,一直莽夫一直爽。

真的想要改得完美一些的话,不仿把这段代码重写一遍。痛点就在这附近。

pooled_height /= 2它不香吗?

(逃)

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值