­

47.4mAP!最强Anchor-free目标检测网络:SAPD

  • 2019 年 12 月 10 日
  • 筆記

前言

最近目标检测相关的论文非常多,可见Object Detection还是有很多工作(坑)可以做(填)的。上次 Amusi 就整理了近期值得关注的目标检测论文,详见:一文看尽8篇目标检测最新论文(EfficientDet/EdgeNet/ASFF/RoIMix/SCL/EFGRNet等)

纵览2019年,目标检测除了NAS分支外,Anchor-free算是占据"半壁江山"。其中比较亮眼的工作有很多,如:FSAF、FCOS、CenterNet、FoveaBox和FreeAnchor等

关于Anchor-free更多的解读,详见下面两篇文章:

目标检测:Anchor-Free时代

重磅!13篇基于Anchor free的目标检测方法

SAPD:Soft Anchor-Point Detector

时间:2019年11月28日

作者团队:卡耐基梅隆大学(CMU)

论文链接:https://arxiv.org/abs/1911.12448

近来,anchor-free 目标检测器在准确性和速度方面都显著的超越 anchor-based 的检测器。在SAPD这个工作中,作者旨在为anchor-free 检测器寻找速度(FPS)和准确性(mAP)之间的新平衡。

Single-model single-scale speed (ms) vs. accuracy (AP) on COCO test-dev

SAPD 研究了两个问题:

1)如何使 anchor-free 目标检测的"head"更好?

2)如何更好地利用特征金字塔的特性?

作者将注意力偏差(attention bias)和特征选择分别确定为上述两个问题的主要痛点。于是作者提出使用一种新的训练策略来解决这些问题,该策略具有两种 soften 优化方法,即:

SW:soft-weighted anchor points

SS:soft-selected pyramid levels

为了评估算法的有效性,作者训练了一个称为SAPD:Soft Anchor-Point Detector的single-stage anchor-free 检测器。实验表明,优雅的 SAPD 将速度/准确度的 trade-off 达到了一个新的水平,超过了之前的 anchor-based,anchor-free,single-stage和 multi-stage 检测器(这里其实都是各家吹捧的点,各种打败SOTA,大家听听就好,因为实际上目前并没有出现mAP和FPS都是SOTA的检测算法)。

SAPD 可以在 COCO 上实现 47.4% mAP(单模型单尺度)。作者称:SAPD最快的版本可以比其他具有相同的精度的检测器速度快5倍!

网络结构

anchor-point detector

训练策略:soft-weighted anchor points 和 soft-selected pyramid levels

实验结果

为了方便下载,我已经将 SAPD 论文打包,在 CVer公众号 后台回复:20191206 即可获得打包链接。同时论文会同步推送到 github上,欢迎大家 star/fork(点击阅读原文,也可直接访问):

https://github.com/amusi/daily-paper-computer-vision

如果各位 CVer 喜欢这样的最强系列,请给这篇文章点个 在看