4 个月前

基于循环尺度近似的CNN目标检测

基于循环尺度近似的CNN目标检测

摘要

由于卷积神经网络(CNN)缺乏处理大规模变化的内在机制,我们在进行多尺度目标检测时总是需要多次计算特征图,这在实际应用中存在计算成本的瓶颈。为了解决这一问题,我们设计了一种递归尺度近似(RSA)方法,仅需计算一次特征图,并通过该图来近似其他层次的特征图。RSA的核心是递归展开机制:给定某一特定尺度下的初始图,它生成一个比输入小一半尺度的预测图。为了进一步提高效率和准确性,我们采取了以下措施:(a) 设计了一个尺度预测网络,用于全局预测图像中的潜在尺度,因为无需在金字塔的所有层次上计算特征图;(b) 提出了一种地标回溯网络(LRN),用于追踪回归地标的实际位置并为每个地标生成置信度分数;LRN可以有效缓解RSA中累积误差导致的误检问题。整个系统可以在统一的CNN框架下进行端到端训练。实验结果表明,我们提出的算法在人脸检测基准测试中优于现有最先进方法,并且在通用提议生成方面也取得了可比的结果。RSA的源代码可在github.com/sciencefans/RSA-for-object-detection获取。

代码仓库

基准测试

基准方法指标
face-detection-on-annotated-faces-in-the-wildLRN + RSA
AP: 0.9917

用 AI 构建 AI

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

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

Hyper Newsletters

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