smooth_L1_loss_layer.cpp:28] Check failed: bottom[0]-channels() == bottom[1]-channels() (12 vs. 84
生活随笔
收集整理的這篇文章主要介紹了
smooth_L1_loss_layer.cpp:28] Check failed: bottom[0]-channels() == bottom[1]-channels() (12 vs. 84
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
用faster RCNN訓(xùn)練自己的數(shù)據(jù)時(shí)候遇到的問題,具體為使用vgg16的end_2_end訓(xùn)練
這種問題一般都是train.prototxt的類別數(shù)沒有改好導(dǎo)致的
在的vgg16的end_2_end下的train.prototxt需要改的是
data層的num_classes: n (自己要訓(xùn)練的類別+1,1代表背景)
以及 roi_data層的num_classes:n
cls_score層的num_output:n
bbox_preda層的num_output : 4*n
有問題歡迎交流
總結(jié)
以上是生活随笔為你收集整理的smooth_L1_loss_layer.cpp:28] Check failed: bottom[0]-channels() == bottom[1]-channels() (12 vs. 84的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: voc_eval.py:41: Runt
- 下一篇: AttributeError: 'mod