yolov3-tiny darknet转caffe模型出现维度不匹配的问题

在运行darknet2caffe-yolov3.py时出现下面报错:

I1205 10:02:01.384008 24139 net.cpp:469] layer21-route <- layer20-upsample
I1205 10:02:01.384047 24139 net.cpp:469] layer21-route <- layer9-conv_layer9-act_0_split_1
I1205 10:02:01.384073 24139 net.cpp:443] layer21-route -> layer21-route
F1205 10:02:01.384135 24139 concat_layer.cpp:42] Check failed: top_shape[j] == bottom[i]->shape(j) (62 vs. 64) All inputs must have the same shape, except at concat_axis

解决方式:
1.首先查看concat连接层的维度,也就是layer20-upsample和layer9-conv_layer9-act_0_split_1,在报错位置向上翻即可找到

可以看到这两层的维度分别为(1,128,62,126)和(1,256,64,128),因此出现了top_shape[j] == bottom[i]->shape(j) (62 vs. 64)报错,这个问题有可能是maxpool参数给错导致的

2.继续顺着报错往上翻,看第二维度62这个数字是从什么地方出现的

可以看到最早是经第12层的最大池化维度从32->31,再经后面的卷积、上采样啥的变成了62

3.虽然出现了报错,但是当前目录应该还是生成了一个prototxt文件,比如我的是yolov3-tiny.prototxt,打开这个文件和你转换的yolov3-tiny.cfg

4.在yolov3-tiny.prototxt找到layer12-maxpool,在yolov3-tiny.cfg找到该maxpool对应的位置,可以看到当前maxpool池化核的大小为2,将其改成1输出就正常了

yolov3-tiny.prototxt
yolov3-tiny.cfg

 5.出现这个问题搜了很多地方没找到合适的解决方法,故记录下来。如果您也出现了类似的错误,可以按照上述思路试试。

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值