OHEM

提出目的是为了解决fastrcnn中正负样本不均衡的问题 算法具体流程如下:先计算出feature map,然后上面的那个ROI网络对选中的B个roi进行前向计算并且计算出每个Roi的损失,将损失排序,然后根据loss执行NMS算法(避免loss大的为同一区域的roi),然后选择hard Rois输入到下面的那个红色的可读可写的ROI网络中,执行前向计算,然后反向传播更新网络参数,最后把这些更新好
相关文章
相关标签/搜索