1. 程式人生 > >yolo_v2訓練自己的資料出現Obj: 0.000000, No Obj: 0.000000

yolo_v2訓練自己的資料出現Obj: 0.000000, No Obj: 0.000000

用yolo_v2訓練自己的資料,,參考各類部落格,準備好了資料,修改了配置檔案,但訓練時不斷出現各種問題。剛開始出現loss為nan的情況,經排查是學習速率太大,loss發散了(因訓練資料較少,約1500張圖片)
再次訓練出現Obj: 0.000000, No Obj: 0.000000的情況,這個問題說明網路沒有檢查到任何物體,截圖如下:
這裡寫圖片描述
解決步驟:
1.上網搜解決方案,有說是資料集問題,但檢查了標籤內容、存放路徑、配置檔案等,沒問題
2.懷疑自己無意修改了yolo-xxx.cfg檔案的某些引數,於是重新複製了yolo-voc.cfg的內容,修改batch=32,max_batches為120次epoch(即,訓練資料量/32*120),最後一個[convolutional]中filters和[region]中的classes。學習速率按自己的實際情況調節,建議慢慢試(太大太小都不行,太大發散,太小易陷入區域性最優)
成功解決~