文章目录
paper: https://arxiv.org/pdf/2012.05780.pdf
code: https://github.com/PeizeSun/OneNet
1. Abstract
文中提出了一个观点,认为以往的one-stage无法去掉NMS实现端到端的主要原因在于正样本的选取。以往的Label assignment任务没有考虑分类的cost,只考虑了位置的cost,从而会得到很多冗余的boxes,造成了后处理中必须使用NMS。
如下图所示,RetinaNet选取正样本只考虑了box的IoU ,而FCOS只考虑了Point Assignment,本文作者提出的OneNet,将classification和location都考虑来选取正样本并计算cost,提出Minimum Cost Assignment方法实现One-to-One Label Assignment,对于每一个Ground Truth,只有一个正样本与之对应,从而获得最佳Match,消除NMS。
2. Contribution
① 网络是端到端的全卷积网络,没有ROI以及attention机制。
②正样本的选取基于对于location