darknet框架下yolov3訓練的常見錯誤

1. 網絡設置錯誤

(1)filter設置

根據你檢測的類型,比如有三類,則yolo輸出節點前的filter值必須設置爲(class+1)*3的大小,如下圖所示:

對應的cfg文件中的第16層以及23層輸出前的filter值,classes也不要忘了修改成對應的數目。否則會報如下錯誤:

parse_region:assertion 'l.output==params.input' failed

voc.data中的classes也要修改。

2. 找不到訓練圖片的路徑

(1)先檢查訓練路徑是否設置正確了;

(2)如使用voc數據格式進行訓練,需檢查“train.txt”文件的編碼格式,一般再windows上生成的txt文件拷貝好linux系統下就會報找不到路徑的問題,這是由編碼方式錯誤造成的。建議在linux下使用命令生成訓練的txt文件,可使用以下命令:

find `pwd`/train -name \*.jpg > train.list

即:在train文件夾下尋找.jpg文件,並在當前目錄下生成train.list文本。實際使用時需要將train修改成自己的圖片文件夾路徑。

3. 內存不足怎麼修改

修改下圖中subdivisions的數值,將該值調大,但必須保持batch/subdivisions爲4的整數倍數。

不定期更新中。。。。。。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章