2 个月前

广义焦点损失V2:学习可靠的密集目标检测定位质量估计

Li, Xiang ; Wang, Wenhai ; Hu, Xiaolin ; Li, Jun ; Tang, Jinhui ; Yang, Jian
广义焦点损失V2:学习可靠的密集目标检测定位质量估计
摘要

定位质量估计(LQE)在近期密集目标检测器的发展中至关重要且广受欢迎,因为它可以提供精确的排名分数,有助于非极大值抑制处理并提升检测性能。目前大多数现有方法通过与目标分类或边界框回归共享的标准卷积特征来预测LQE分数。本文中,我们探索了一种全新的视角来进行LQE——基于边界框的四个参数所学习到的分布。这些边界框分布受到GFLV1中“通用分布”概念的启发和引入,该概念能够很好地描述预测边界框的不确定性。这种特性使得边界框的分布统计与其实际定位质量高度相关。具体而言,具有尖锐峰值的边界框分布通常对应较高的定位质量,反之亦然。通过利用分布统计与实际定位质量之间的紧密关联,我们开发了一个相当轻量级的分布引导质量预测器(DGQP),以基于GFLV1实现可靠的LQE,从而生成了GFLV2。据我们所知,这是首次在目标检测中尝试使用高度相关的统计表示来促进LQE。大量实验验证了我们方法的有效性。特别是,GFLV2(ResNet-101)在COCO {\tt test-dev}数据集上实现了46.2 AP(每秒14.6帧),比之前的最先进ATSS基线(43.6 AP每秒14.6帧)绝对提升了2.6 AP,同时没有牺牲训练和推理过程中的效率。代码将在https://github.com/implus/GFocalV2发布。