Instant 3D Object Tracking with Applications in Augmented Reality

TL;DR

Proposed a calibration-free, real-time 3D object tracking system combining neural detection and planar tracking on mobile devices, achieving 26FPS.

cs.CV 🔴 Advanced 2020-06-24 45 views
Adel Ahmadyan Tingbo Hou Jianing Wei Liangkai Zhang Artsiom Ablavatski Matthias Grundmann
3D tracking augmented reality deep learning mobile real-time

Key Findings

Methodology

The system integrates a lightweight CNN based on MobileNetV2 for object detection and initial 3D pose estimation. It predicts 3D bounding boxes directly from RGB images, then projects box vertices onto the image plane. Using a robust planar tracker, it tracks nine keypoints (eight vertices plus center) across frames. The tracked 2D points are lifted back into 3D space via the EPnP algorithm, enabling continuous 9-DoF pose estimation. Efficient CPU-GPU collaboration ensures real-time performance (26FPS detection, 30+FPS tracking) on mobile devices.

Key Results

  • The detection network achieves an average precision of 0.59 at 3D IoU threshold 0.5, with a model size of only 5.54MB. On a Samsung S20 with Snapdragon 865, detection runs at 26.5FPS, and tracking exceeds 30FPS on CPU, demonstrating high efficiency.
  • In multi-object scenarios, the system maintains stable, continuous tracking across categories like shoes and chairs, outperforming traditional geometric or deep learning-only methods, especially under dynamic conditions.
  • Ablation studies confirm that combining detection with tracking reduces jitter and improves stability, validating the robustness of the integrated approach.

Significance

This work advances mobile AR by providing a simple, calibration-free, and real-time 3D tracking solution. It addresses key limitations of prior methods relying on depth sensors or complex initialization, enabling broader adoption in consumer devices. Its ability to handle unseen objects and operate without prior models opens new avenues for AR applications such as virtual try-on, furniture placement, and interactive gaming, significantly lowering hardware barriers and expanding industry impact.

Technical Contribution

The core innovation lies in combining a neural network-based detection module with geometric pose estimation (EPnP) and a planar tracker, creating a seamless, end-to-end pipeline. This approach avoids the need for pre-calibration or depth sensors, while leveraging efficient resource utilization on mobile hardware. The system supports full 9-DoF tracking at high frame rates, surpassing existing solutions in simplicity and performance.

Novelty

This is the first system to achieve calibration-free, real-time 3D object tracking solely from monocular RGB input, integrating deep learning detection with geometric pose estimation and planar tracking. Unlike prior works dependent on depth sensors or multi-view setups, this approach simplifies hardware requirements and enhances applicability in consumer AR devices.

Limitations

  • The system assumes the target object remains on a planar surface; significant roll or non-planar movements can cause tracking failure.
  • Symmetric objects or those with ambiguous geometry may result in inconsistent orientation estimates, especially for objects like volleyball.
  • Rapid motion, occlusion, or complex backgrounds can reduce robustness, requiring further improvements for challenging scenarios.

Future Work

Future research will focus on extending support to non-planar and articulated objects, improving robustness against occlusion and fast motion. Incorporating multi-frame temporal models and sensor fusion (e.g., IMU) could further enhance stability. Additionally, optimizing models for lower latency and broader category generalization remains a key goal.

AI Executive Summary

Achieving accurate and real-time 3D object tracking on mobile devices remains a significant challenge in augmented reality. Traditional approaches often rely on depth sensors or complex calibration procedures, limiting their practicality for consumer applications. This paper introduces an innovative system that leverages deep learning and geometric algorithms to perform instant 3D pose estimation from monocular RGB images, without any prior calibration or model knowledge.

The core of the system is a lightweight CNN based on MobileNetV2, trained to detect objects and estimate their initial 3D bounding boxes. These boxes are then projected onto the image plane, and a robust planar tracker follows nine keypoints (eight vertices plus center) across frames. The tracked 2D points are lifted back into 3D space using the EPnP algorithm, enabling continuous 9-DoF pose estimation. This combination allows the system to operate efficiently on mobile hardware, achieving 26FPS for detection and over 30FPS for tracking.

Experimental results demonstrate that the system maintains high accuracy, with an average 3D IoU of 0.59 at 0.5 threshold, across categories like shoes and chairs. The small model size (5.54MB) and high frame rate validate its suitability for real-world AR applications such as virtual try-on, furniture placement, and interactive gaming. The approach's simplicity, robustness, and category generalization mark a significant step forward in mobile AR technology.

The authors highlight that their system does not require complex calibration, prior shape models, or depth sensors, greatly lowering the barrier for widespread deployment. Its ability to handle multiple objects simultaneously and maintain stability under dynamic conditions underscores its practical value. Future directions include expanding support for non-planar objects, improving robustness under occlusion, and integrating sensor fusion techniques to further enhance performance and applicability.

Deep Analysis

Background

目标追踪在AR和机器人领域一直是研究热点。早期方法如光流和特征匹配在静态环境中表现良好,但在动态场景中鲁棒性不足。深度学习的引入带来了显著突破,例如Mask R-CNN和PoseCNN,但多依赖深度传感器或复杂初始化。单目追踪因其硬件成本低、应用广泛而受到关注,但受尺度模糊、遮挡和快速运动影响,难以实现连续、准确的全自由度追踪。近年来,结合几何和学习的混合方法逐渐兴起,但仍存在实时性和鲁棒性不足的问题。本文旨在利用深度学习检测和几何算法结合,提供一种高效、无需校准的移动端3D追踪方案。

Core Problem

核心问题是如何在没有深度信息和复杂初始化的情况下,实现单目RGB图像中目标的连续、精确的3D追踪。传统方法依赖深度传感器或多视角信息,成本高且不易普及。单目追踪面临尺度模糊、姿态不唯一、遮挡干扰等挑战,尤其在移动设备算力有限的条件下,如何保证追踪的鲁棒性和实时性成为难点。解决这一问题需要结合深度学习的检测能力与几何估计的精确性,设计高效的追踪策略。

Innovation

本研究的创新包括:1)提出无需预设模型或校准的端到端追踪系统,结合深度学习检测与EPnP几何估算,实现目标的初始3D姿态;2)利用平面追踪器持续追踪目标在图像中的九个关键点,结合相对尺度模型实现连续9-DoF追踪;3)充分利用移动设备的CPU和GPU资源,确保系统在保持高帧率的同时,具有良好的鲁棒性。这一方案突破了传统依赖深度传感器和复杂初始化的限制,简化了系统架构。

Methodology

  • �� 输入:单帧RGB图像。• 目标检测:利用基于MobileNetV2的多任务网络同时预测目标类别、边界框和形状信息(如分割掩码)。• 初始姿态:通过检测输出的边界框,拟合高斯分布确定目标中心位置,结合边界框顶点投影,使用EPnP算法估算目标空间位置和尺度。• 关键点投影:计算边界框的8个顶点和中心点,将其投影到图像平面。• 追踪:采用平面追踪器持续追踪这9个2D点,结合相对尺度的9-DoF模型,保持连续追踪。• 3D重建:每帧将2D点反投影到3D空间,利用EPnP更新目标的空间位置和姿态。• 融合:检测与追踪结果通过区域重叠进行融合,提升稳定性。

Experiments

使用自制鞋子和椅子数据集,评估检测精度和追踪鲁棒性。对比传统几何追踪和纯深度学习方法,验证检测+追踪方案在动态环境中的优势。指标包括3D IoU、帧率和鲁棒性。在不同光照、遮挡和快速运动条件下测试,确保系统稳定。训练采用多任务损失,追踪模型基于相对尺度,优化实时性能和准确性。

Results

检测模型在3D IoU阈值0.5时达0.59平均精度,模型大小5.54MB,在移动GPU上实现26.5FPS检测,追踪在CPU上超过30FPS。多目标追踪表现稳定,尤其在动态场景中优于传统方法。消融实验显示,检测+追踪策略显著减少抖动,提升稳定性,验证了方案的有效性。

Applications

该系统适用于虚拟试衣、家具摆放、增强现实游戏等。只需普通摄像头,无需预先模型或校准,即可实现实时追踪。其高效性和类别泛化能力,为移动端AR带来巨大潜力,降低技术门槛,推动产业普及。

Limitations & Outlook

系统假设目标在追踪过程中保持平面,若目标发生旋转或滚转,可能导致追踪失败。对对称或复杂几何结构的对象,可能出现姿态不唯一或误差增加。在快速运动或遮挡严重的场景中,鲁棒性仍需提升。未来需解决非平面目标、多目标同时追踪等挑战。

Plain Language Accessible to non-experts

想象你在厨房里做饭,锅里的食材不断变化。传统方法像用手电筒照明,只能看到表面,但不能知道食材的具体位置和大小。现在,这个系统像装了一个聪明的摄像头,能自动识别锅里的食材,并且知道它们在空间中的位置、大小和方向。它不需要提前告诉系统食材长什么样,也不用复杂的校准,只用一部普通手机就能做到。系统先用AI识别目标,然后用几何方法估算它在空间中的位置,接着用平面追踪器持续跟踪它的运动。这样,无论你在厨房里走动多快,系统都能实时跟踪食材的变化,就像有个聪明的助手帮你盯着锅里的食物一样。这个技术可以用在虚拟试衣、家具摆放等场景,让虚拟内容和真实物体完美结合,带来更自然的AR体验。

ELI14 Explained like you're 14

想象你在玩一个游戏,你的手机可以看见你房间里的玩具,然后帮你把虚拟的机器人放在玩具旁边,好像它们真的在一起一样。这就像手机变成了一个超级聪明的眼睛,能看见房间里的东西,并且知道它们在空间里的位置和方向。这个系统不用特殊设备,只用普通的手机摄像头,就能做到。它先用一个聪明的程序找出你想追踪的东西,比如一把椅子,然后用数学方法估算出椅子在房间里的具体位置和大小。接着,它会一直跟踪椅子,确保虚拟的东西可以和真实物体完美结合。这样,你可以用手机做虚拟试穿、摆放家具,甚至玩增强现实游戏,体验更真实、更有趣的虚拟世界。这个技术让AR变得更简单、更快,也更适合普通人使用。

Glossary

3D边界框(3D Bounding Box)

在三维空间中,用长宽高和姿态描述目标的包围盒,代表目标的空间位置和大小。

用于初始化目标的空间位置和尺度。

EPnP(Efficient Perspective-n-Point)

一种快速求解相机姿态的几何算法,通过已知3D点和对应2D投影点估算相机位置与方向。

用于从边界框顶点投影中估算目标的3D姿态。

平面追踪器(Planar Tracker)

基于图像平面特征持续追踪目标的二维关键点,保持目标在视频中的连续性。

实现目标在连续帧中的稳定追踪。

9-DoF(九自由度)

目标的三维位置(3个自由度)、姿态(3个自由度)和尺度(3个自由度),描述目标在空间中的完整运动状态。

实现目标的全自由度追踪。

深度神经网络(Deep Neural Network)

由多层神经元组成的模型,用于学习复杂的特征表示,广泛应用于目标检测和姿态估计。

实现目标的初始检测和姿态估计。

Open Questions Unanswered questions from this research

  • 1 在复杂环境中,目标遮挡或快速运动对追踪的影响仍未充分解决,如何提升鲁棒性和连续性是未来的关键问题。
  • 2 系统在非平面目标或多目标同时追踪方面的能力仍有限,需进一步研究多目标协同追踪算法。

Abstract

Tracking object poses in 3D is a crucial building block for Augmented Reality applications. We propose an instant motion tracking system that tracks an object's pose in space (represented by its 3D bounding box) in real-time on mobile devices. Our system does not require any prior sensory calibration or initialization to function. We employ a deep neural network to detect objects and estimate their initial 3D pose. Then the estimated pose is tracked using a robust planar tracker. Our tracker is capable of performing relative-scale 9-DoF tracking in real-time on mobile devices. By combining use of CPU and GPU efficiently, we achieve 26-FPS+ performance on mobile devices.

cs.CV