arXiv:2412.13273cs.CV2024-12被引 2

CompactFlowNet让手机实时计算画面运动轨迹,精度不降速度飞升。

CompactFlowNet: Efficient Real-time Optical Flow Estimation on Mobile Devices

  • 设计轻量架构+优化训练流程,专为手机端低延迟高效率定制。
  • 在KITTI和Sintel数据集上表现优于或媲美现有轻量模型。
  • iPhone 8上实现实时推理,高端手机更超速运行,适合移动端视频应用。

我们提出CompactFlowNet,首个专为移动设备设计的实时光流预测神经网络,旨在确定初始帧中每个像素相对于后续帧对应像素的位移。光流是视频修复、运动估计、视频稳定、目标跟踪、动作识别和视频生成等任务的基础。当前先进方法虽追求高精度,却常忽略速度与内存限制;现有轻量模型通常仅缩减规模,仍存在高延迟、质量显著下降或仅适配高性能GPU的问题,导致移动端表现不佳。本研究通过提出新型移动端兼容架构及训练流程优化,使模型在保持极低误差的同时实现更小参数量、更低内存占用和更高推理速度。实验表明,CompactFlowNet在挑战性KITTI和Sintel基准上性能优于或媲美现有最先进轻量模型,并在iPhone 8上实现实时推理,在更高级移动设备上表现远超实时标准。

原文摘要 · Abstract (English)

We present CompactFlowNet, the first real-time mobile neural network for optical flow prediction, which involves determining the displacement of each pixel in an initial frame relative to the corresponding pixel in a subsequent frame. Optical flow serves as a fundamental building block for various video-related tasks, such as video restoration, motion estimation, video stabilization, object tracking, action recognition, and video generation. While current state-of-the-art methods prioritize accuracy, they often overlook constraints regarding speed and memory usage. Existing light models typically focus on reducing size but still exhibit high latency, compromise significantly on quality, or are optimized for high-performance GPUs, resulting in sub-optimal performance on mobile devices. This study aims to develop a mobile-optimized optical flow model by proposing a novel mobile device-compatible architecture, as well as enhancements to the training pipeline, which optimize the model for reduced weight, low memory utilization, and increased speed while maintaining minimal error. Our approach demonstrates superior or comparable performance to the state-of-the-art lightweight models on the challenging KITTI and Sintel benchmarks. Furthermore, it attains a significantly accelerated inference speed, thereby yielding real-time operational efficiency on the iPhone 8, while surpassing real-time performance levels on more advanced mobile devices.

光流估计移动端实时推理轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。