OHEM

提出目的是爲了解決fastrcnn中正負樣本不均衡的問題 算法具體流程如下:先計算出feature map,然後上面的那個ROI網絡對選中的B個roi進行前向計算並且計算出每個Roi的損失,將損失排序,然後根據loss執行NMS算法(避免loss大的爲同一區域的roi),然後選擇hard Rois輸入到下面的那個紅色的可讀可寫的ROI網絡中,執行前向計算,然後反向傳播更新網絡參數,最後把這些更新好
相關文章
相關標籤/搜索