提出目的是为了解决fastrcnn中正负样本不均衡的问题

算法具体流程如下:先计算出feature map,然后上面的那个ROI网络对选中的B个roi进行前向计算并且计算出每个Roi的损失,将损失排序,然后根据loss执行NMS算法(避免loss大的为同一区域的roi),然后选择hard Rois输入到下面的那个红色的可读可写的ROI网络中,执行前向计算,然后反向传播更新网络参数,最后把这些更新好的参数传给上面的那个绿色的只读的ROI网络,即一次迭代完成。

OHEM

也就是说网络有两个head,其中一个head只计算前向传播的loss,并根据LOSS进行NMS,这样在另一个head就可以有充足的hard example进行训练

https://zhuanlan.zhihu.com/p/85815803

相关文章:

  • 2022-12-23
  • 2021-11-01
  • 2021-05-12
  • 2021-07-31
  • 2022-01-07
  • 2021-11-23
  • 2021-06-21
猜你喜欢
  • 2022-12-23
  • 2021-05-03
  • 2021-06-12
  • 2022-01-14
相关资源
相似解决方案