天天看點

用資料集跑一個模型遇到bug如何解決

自己在用fast rcnn和ssd跑自己資料集過程中都遇到了bug,fast rcnn中是loss下降但值較高,并且測試出來結果一直不對,ssd是loss從一開始到後面loss都一直為0。

遇到這種情況,最好是先把資料集換成一張圖去跑,并且定位bug。

更明顯的例子,自己在fast rcnn中,修改了坐标的那個bug,感覺ap值應該很高,但ap值依舊很低,後來發現是ap實作代碼出錯。當時自己使用一個圖檔的資料進行ap測試,發現結果是正确的,但多個就錯了,進而在後面定位了代碼在哪裡出錯。