arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.02574cs.CVcs.LG
收藏

时空提示网络用于鲁棒视频特征提取

Spatio-temporal Prompting Network for Robust Video Feature Extraction

Huawei UKRD(华为英国研发部);Imperial College London(伦敦帝国学院)

Guanxiong Sun, Chi Wang, Zhaoyu Zhang, Jiankang Deng, Stefanos Zafeiriou, Yang Hua

AI总结:

本文提出时空提示网络(STPN),通过动态调整输入特征并前置视频提示,高效提取鲁棒视频特征,在多个视频任务数据集上达到最先进性能。

摘要:

帧质量退化是视频理解领域的主要挑战之一。为补偿退化帧造成的信息损失,近期方法利用基于Transformer的集成模块来获取时空信息。然而,这些集成模块既笨重又复杂。此外,每个集成模块都是针对其目标任务专门设计的,难以泛化到多个任务。在本文中,我们提出了一个简洁且统一的框架,称为时空提示网络(STPN)。它通过动态调整骨干网络中的输入特征,能够高效地提取鲁棒且准确的视频特征。具体而言,STPN预测多个包含相邻帧时空信息的视频提示。然后,将这些视频提示前置到当前帧的补丁嵌入中,作为视频特征提取的更新输入。此外,由于STPN不包含任务特定模块,因此易于泛化到各种视频任务。无需额外花哨操作,STPN在三个广泛使用的不同视频理解任务数据集上达到了最先进的性能,即用于视频目标检测的ImageNetVID、用于视频实例分割的YouTubeVIS以及用于视觉目标跟踪的GOT-10k。代码可在https://github.com/guanxiongsun/vfe.pytorch获取。

英文摘要:

Frame quality deterioration is one of the main challenges in the field of video understanding. To compensate for the information loss caused by deteriorated frames, recent approaches exploit transformer-based integration modules to obtain spatio-temporal information. However, these integration modules are heavy and complex. Furthermore, each integration module is specifically tailored for its target task, making it difficult to generalise to multiple tasks. In this paper, we present a neat and unified framework, called Spatio-Temporal Prompting Network (STPN). It can efficiently extract robust and accurate video features by dynamically adjusting the input features in the backbone network. Specifically, STPN predicts several video prompts containing spatio-temporal information of neighbour frames. Then, these video prompts are prepended to the patch embeddings of the current frame as the updated input for video feature extraction. Moreover, STPN is easy to generalise to various video tasks because it does not contain task-specific modules. Without bells and whistles, STPN achieves state-of-the-art performance on three widely-used datasets for different video understanding tasks, i.e., ImageNetVID for video object detection, YouTubeVIS for video instance segmentation, and GOT-10k for visual object tracking. Code is available at https://github.com/guanxiongsun/vfe.pytorch.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑