基於源碼的Faster R-CNN訓練總體流程梳理

之前的文章中對於Faster R-CNN的總體流程進行了梳理,詳情參見這裏。本篇文章將要以源碼中的結構爲基礎,對於訓練的整體流程進行梳理,以便後續源碼的閱讀。 首先我們要明確的是,實際的Faster R-CNN中訓練分爲6個步驟: 第一次訓練RPN網絡,對應源碼中的stage1_rpn_train.pt 利用步驟1訓練好的RPN網絡,收集proposals,對應rpn_test.pt 第一次訓練F
相關文章
相關標籤/搜索