4 个月前

软锚点目标检测

软锚点目标检测

摘要

近年来,无锚点检测方法取得了显著进展。主要分为两大类:无锚点检测和关键点检测,它们在速度与精度的权衡上处于两个极端,其中无锚点检测器在速度方面具有优势。在这项工作中,我们在保持速度优势的同时,提升了无锚点检测器相对于关键点检测器的性能。为了实现这一目标,我们从无锚点的角度重新定义了检测问题,并确定无效训练是主要问题。我们的核心见解是,无锚点应该在特征金字塔内部和跨层级之间联合优化。为此,我们提出了一种简单而有效的训练策略,即软加权无锚点(soft-weighted anchor points)和软选择金字塔层级(soft-selected pyramid levels),分别解决了每个金字塔层级内的虚假注意力问题和跨所有金字塔层级的特征选择问题。为了评估该策略的有效性,我们训练了一个名为软无锚点检测器(Soft Anchor-Point Detector, SAPD)的单阶段无锚点检测器。实验结果表明,我们的简洁SAPD将速度与精度的权衡提升到了一个新的水平,优于最近的最先进的无锚点和有锚点检测器。无需复杂的技巧,我们的最佳模型在COCO数据集上可以达到47.4%的单模型单尺度平均精度(AP)。

代码仓库

xuannianz/FSAF
tf
GitHub 中提及
xuannianz/SAPD
tf
GitHub 中提及

基准测试

基准方法指标
dense-object-detection-on-sku-110kSAPD
AP: 55.7
object-detection-on-cocoSAPD (ResNeXt-101, single-scale)
AP50: 67.4
AP75: 51.1
APL: 61.5
APM: 50.3
APS: 28.1
Hardware Burden:
Operations per network pass:
box mAP: 47.4

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
软锚点目标检测 | 论文 | HyperAI超神经