Saved in:
Bibliographic Details
Main Authors: Znobishchev, Andrei, Filev, Valerii, Kudashev, Oleg, Orlov, Nikita, Shi, Humphrey
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2412.13273
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912160189251584
author Znobishchev, Andrei
Filev, Valerii
Kudashev, Oleg
Orlov, Nikita
Shi, Humphrey
author_facet Znobishchev, Andrei
Filev, Valerii
Kudashev, Oleg
Orlov, Nikita
Shi, Humphrey
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2412_13273
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CompactFlowNet: Efficient Real-time Optical Flow Estimation on Mobile Devices
Znobishchev, Andrei
Filev, Valerii
Kudashev, Oleg
Orlov, Nikita
Shi, Humphrey
Computer Vision and Pattern Recognition
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.
title CompactFlowNet: Efficient Real-time Optical Flow Estimation on Mobile Devices
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.13273