OneVOS: Unifying Video Object Segmentation with All-in-One Transformer Framework

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Wanyun, Guo, Pinxue, Zhou, Xinyu, Hong, Lingyi, He, Yangji, Zheng, Xiangyu, Zhang, Wei, Zhang, Wenqiang
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929275092860928
author Li, Wanyun
Guo, Pinxue
Zhou, Xinyu
Hong, Lingyi
He, Yangji
Zheng, Xiangyu
Zhang, Wei
Zhang, Wenqiang
author_facet Li, Wanyun
Guo, Pinxue
Zhou, Xinyu
Hong, Lingyi
He, Yangji
Zheng, Xiangyu
Zhang, Wei
Zhang, Wenqiang
contents Contemporary Video Object Segmentation (VOS) approaches typically consist stages of feature extraction, matching, memory management, and multiple objects aggregation. Recent advanced models either employ a discrete modeling for these components in a sequential manner, or optimize a combined pipeline through substructure aggregation. However, these existing explicit staged approaches prevent the VOS framework from being optimized as a unified whole, leading to the limited capacity and suboptimal performance in tackling complex videos. In this paper, we propose OneVOS, a novel framework that unifies the core components of VOS with All-in-One Transformer. Specifically, to unify all aforementioned modules into a vision transformer, we model all the features of frames, masks and memory for multiple objects as transformer tokens, and integrally accomplish feature extraction, matching and memory management of multiple objects through the flexible attention mechanism. Furthermore, a Unidirectional Hybrid Attention is proposed through a double decoupling of the original attention operation, to rectify semantic errors and ambiguities of stored tokens in OneVOS framework. Finally, to alleviate the storage burden and expedite inference, we propose the Dynamic Token Selector, which unveils the working mechanism of OneVOS and naturally leads to a more efficient version of OneVOS. Extensive experiments demonstrate the superiority of OneVOS, achieving state-of-the-art performance across 7 datasets, particularly excelling in complex LVOS and MOSE datasets with 70.1% and 66.4% $J \& F$ scores, surpassing previous state-of-the-art methods by 4.2% and 7.0%, respectively. And our code will be available for reproducibility and further research.
format Preprint
id arxiv_https___arxiv_org_abs_2403_08682
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle OneVOS: Unifying Video Object Segmentation with All-in-One Transformer Framework
Li, Wanyun
Guo, Pinxue
Zhou, Xinyu
Hong, Lingyi
He, Yangji
Zheng, Xiangyu
Zhang, Wei
Zhang, Wenqiang
Computer Vision and Pattern Recognition
Contemporary Video Object Segmentation (VOS) approaches typically consist stages of feature extraction, matching, memory management, and multiple objects aggregation. Recent advanced models either employ a discrete modeling for these components in a sequential manner, or optimize a combined pipeline through substructure aggregation. However, these existing explicit staged approaches prevent the VOS framework from being optimized as a unified whole, leading to the limited capacity and suboptimal performance in tackling complex videos. In this paper, we propose OneVOS, a novel framework that unifies the core components of VOS with All-in-One Transformer. Specifically, to unify all aforementioned modules into a vision transformer, we model all the features of frames, masks and memory for multiple objects as transformer tokens, and integrally accomplish feature extraction, matching and memory management of multiple objects through the flexible attention mechanism. Furthermore, a Unidirectional Hybrid Attention is proposed through a double decoupling of the original attention operation, to rectify semantic errors and ambiguities of stored tokens in OneVOS framework. Finally, to alleviate the storage burden and expedite inference, we propose the Dynamic Token Selector, which unveils the working mechanism of OneVOS and naturally leads to a more efficient version of OneVOS. Extensive experiments demonstrate the superiority of OneVOS, achieving state-of-the-art performance across 7 datasets, particularly excelling in complex LVOS and MOSE datasets with 70.1% and 66.4% $J \& F$ scores, surpassing previous state-of-the-art methods by 4.2% and 7.0%, respectively. And our code will be available for reproducibility and further research.
title OneVOS: Unifying Video Object Segmentation with All-in-One Transformer Framework
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.08682