Video Instance Segmentation using Inter-Frame Communication Transformers
About
We propose a novel end-to-end solution for video instance segmentation (VIS) based on transformers. Recently, the per-clip pipeline shows superior performance over per-frame methods leveraging richer information from multiple frames. However, previous per-clip models require heavy computation and memory usage to achieve frame-to-frame communications, limiting practicality. In this work, we propose Inter-frame Communication Transformers (IFC), which significantly reduces the overhead for information-passing between frames by efficiently encoding the context within the input clip. Specifically, we propose to utilize concise memory tokens as a mean of conveying information as well as summarizing each frame scene. The features of each frame are enriched and correlated with other frames through exchange of information between the precisely encoded memory tokens. We validate our method on the latest benchmark sets and achieved the state-of-the-art performance (AP 44.6 on YouTube-VIS 2019 val set using the offline inference) while having a considerably fast runtime (89.4 FPS). Our method can also be applied to near-online inference for processing a video in real-time with only a small delay. The code will be made available.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Instance Segmentation | COCO 2017 (val) | -- | 1144 | |
| Video Instance Segmentation | YouTube-VIS 2019 (val) | AP44.6 | 567 | |
| Video Instance Segmentation | YouTube-VIS 2021 (val) | AP36.8 | 344 | |
| Video Instance Segmentation | OVIS (val) | AP13.1 | 301 | |
| Video Instance Segmentation | YouTube-VIS 2019 | AP42.8 | 75 | |
| Video Instance Segmentation | YouTube-VIS 2021 | AP36.6 | 63 | |
| Video Instance Segmentation | YTVIS 2019 (test val) | AP42.8 | 28 | |
| Video Instance Segmentation | HQ-YTVIS (test) | APB27.2 | 20 | |
| Video Instance Segmentation | OVIS 2021 (val) | AP13.1 | 14 |