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
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
用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
有問題歡迎交流
總結
以上是生活随笔為你收集整理的smooth_L1_loss_layer.cpp:28] Check failed: bottom[0]-channels() == bottom[1]-channels() (12 vs. 84的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: voc_eval.py:41: Runt
- 下一篇: AttributeError: 'mod