使用faster-rcnn.pytorch訓練自己數據集(簡潔版)


1、從這里下載FasterRCNN的Pytorch版本代碼。按照提示完成相關編譯。

2、因為默認的數據為VOC2007,最直接的方法我們可以按照這里將自己的數據制作成VOC2007的格式,並建立軟連接。

3、修改pascal_voc.py數據的類別,即將self._classes修改為自己的類。注:保留background類,用於訓練RPN網絡,視作RPN網絡的負樣本。

4、運行trainval_net.py即可。

5、測試時,需要將test_net.py的類別做相應修改。

 

注:① 在每次訓練前,需要將data下面的cache刪掉。

       ② 如果跑demo.py會發現檢測不完整,這是因為demo里最多顯示10個檢測框,修改響應代碼即可。
---------------------
ref:https://blog.csdn.net/yutingzhaomeng/article/details/80328422


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM