arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.11528cs.CV
收藏

MimicDet:缩小一阶段和两阶段目标检测之间的差距

MimicDet: Bridging the Gap Between One-Stage and Two-Stage Object Detection

SenseTime Research(商汤科技研究院);The Chinese University of Hong Kong(香港中文大学)

Xin Lu, Quanquan Li, Buyu Li, Junjie Yan

摘要:

现代目标检测方法可以分为一阶段方法和两阶段方法。由于结构简单,单级检测器效率更高,但两级检测器在精度上仍然领先。虽然最近的工作试图通过模仿两级探测器的结构设计来改进一级探测器,但精度差距仍然很大。在本文中,我们提出了MimicDet,一个新颖的和有效的框架,训练一个阶段的检测器直接模仿两个阶段的功能,旨在弥合之间的准确性差距的一个阶段和两个阶段的检测器。与传统的模拟方法不同,MimicDet具有一级和两级检测器的共享主干,然后它分支成两个头部,这两个头部被精心设计为具有兼容的模拟功能。因此,MimicDet可以在没有教师网络预训练的情况下进行端到端训练。而且成本增加不多,这使得采用大型网络作为骨干网成为现实。我们还提出了一些特殊的设计,如双路径模仿和交错特征金字塔,以方便模仿过程。在具有挑战性的COCO检测基准上的实验证明了MimicDet的有效性。它在COCO测试开发集上使用ResNeXt-101主干实现了46.1 mAP,大大超过了当前最先进的方法。

英文摘要:

Modern object detection methods can be divided into one-stage approaches and two-stage ones. One-stage detectors are more efficient owing to straightforward architectures, but the two-stage detectors still take the lead in accuracy. Although recent work try to improve the one-stage detectors by imitating the structural design of the two-stage ones, the accuracy gap is still significant. In this paper, we propose MimicDet, a novel and efficient framework to train a one-stage detector by directly mimic the two-stage features, aiming to bridge the accuracy gap between one-stage and two-stage detectors. Unlike conventional mimic methods, MimicDet has a shared backbone for one-stage and two-stage detectors, then it branches into two heads which are well designed to have compatible features for mimicking. Thus MimicDet can be end-to-end trained without the pre-train of the teacher network. And the cost does not increase much, which makes it practical to adopt large networks as backbones. We also make several specialized designs such as dual-path mimicking and staggered feature pyramid to facilitate the mimicking process. Experiments on the challenging COCO detection benchmark demonstrate the effectiveness of MimicDet. It achieves 46.1 mAP with ResNeXt-101 backbone on the COCO test-dev set, which significantly surpasses current state-of-the-art methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑