1. 程式人生 > >faster-rcnn訓練自己的資料時報錯too many indices

faster-rcnn訓練自己的資料時報錯too many indices

利用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都不為空