中文

Dynamic R-CNN:通过动态训练实现高质量目标检测

计算机视觉与模式识别 2020-07-28 v2

摘要

尽管两阶段目标检测器近年来不断刷新最优性能,但其训练过程本身远非清晰。本文中,我们首先指出固定网络设置与动态训练过程之间的不一致问题,其极大影响了性能。例如,固定的标签分配策略和回归损失函数无法适应候选框分布的变化,从而不利于训练高质量检测器。为此,我们提出 Dynamic R-CNN,根据训练过程中候选框的统计信息自动调整标签分配标准(IoU 阈值)和回归损失函数的形状(SmoothL1 Loss 的参数)。这种动态设计更好地利用了训练样本,并推动检测器拟合更多高质量样本。具体而言,我们的方法在 MS COCO 数据集上基于 ResNet-50-FPN 基线提升了 1.9% AP 和 5.5% AP90_{90},且无额外开销。代码和模型见 https://github.com/hkzhang95/DynamicRCNN。

关键词

引用

@article{arxiv.2004.06002,
  title  = {Dynamic R-CNN: Towards High Quality Object Detection via Dynamic Training},
  author = {Hongkai Zhang and Hong Chang and Bingpeng Ma and Naiyan Wang and Xilin Chen},
  journal= {arXiv preprint arXiv:2004.06002},
  year   = {2020}
}

备注

Accepted by ECCV2020