HyperAIHyperAI

Command Palette

Search for a command to run...

概率两阶段检测

Xingyi Zhou Vladlen Koltun Philipp Krähenbühl

摘要

我们提出了一种对两阶段目标检测的概率化解释。研究表明,这一概率视角能够解释多种常见的经验性训练实践,并为两阶段检测流程的改进提供新思路。具体而言,第一阶段应准确估计目标与背景的似然概率,该概率应进一步用于决定检测器的综合得分。传统的区域建议网络(RPN)难以充分建模此类似然,而许多单阶段检测器则具备这一能力。我们展示了如何基于任意先进的单阶段检测器构建一个概率化的两阶段检测器。所提出的检测器在速度和精度上均优于其对应的单阶段与两阶段基线模型。在仅使用单尺度测试的情况下,该检测器在COCO test-dev数据集上取得了56.4 mAP的性能,超越了所有已发表的结果。此外,采用轻量级主干网络时,该检测器在Titan Xp上以33 fps的推理速度实现了49.2 mAP的COCO性能,显著优于广泛使用的YOLOv4模型。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供