faster-rcnn訓練自己的資料時報錯too many indices
阿新 • • 發佈:2019-02-04
利用faster-rcnn 好不容易四個階段訓練完了,在測試的時候出現這個錯誤。
原因:在data/VOCdevkit2007/results/VOC2007/Main/ 這個檔案下有n個txt檔案。n是你的類別數
每個txt儲存了test.txt裡面測試集的測試結果,如果某個txt是空的,就會報這種錯。也就是說所有的txt裡面必須有資料
如果誰python學得好,加個判斷語句判斷一下,如果txt為空,跳過,也能解決這個問題。可惜我不會
按照作者原始方法,可以保證每個txt都有資料,而我們自己進行訓練,有時訓練次數少,或者資料少,導致學習的不好,進行測試時,低於 __C.TEST.RPN_NMS_THRESH (預設是0.7)的測試結果被丟掉,所以為了確保每個每個txt都有測試的資料,可以按照一下方法
解決:第一種方法:增加test.txt中的樣本量
第二種方法:/home/zc/py-20-ceshi/lib/fast_rcnn/config.py 把這個值 __C.TEST.RPN_NMS_THRESH 改低一點,具體多少,得看你的類別多少,資料量多少,
學習的好不好
建議兩種方法同時使用,確保所有txt都不為空