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