smooth_L1_loss_layer.cpp:28] Check failed: bottom[0]->channels() == bottom[1]->channels() (12 vs. 84

用faster RCNN训练自己的数据时候遇到的问题,具体为使用vgg16的end_2_end训练

这种问题一般都是train.prototxt的类别数没有改好导致的

在的vgg16的end_2_end下的train.prototxt需要改的是

data层的num_classes: n (自己要训练的类别+1,1代表背景)

以及 roi_data层的num_classes:n

cls_score层的num_output:n

bbox_preda层的num_output : 4*n

有问题欢迎交流

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值