Learning Quality-aware Dynamic Memory for Video Object Segmentation

16 Jul 2022  ·  Yong liu, Ran Yu, Fei Yin, Xinyuan Zhao, Wei Zhao, Weihao Xia, Yujiu Yang ·

Recently, several spatial-temporal memory-based methods have verified that storing intermediate frames and their masks as memory are helpful to segment target objects in videos. However, they mainly focus on better matching between the current frame and the memory frames without explicitly paying attention to the quality of the memory. Therefore, frames with poor segmentation masks are prone to be memorized, which leads to a segmentation mask error accumulation problem and further affect the segmentation performance. In addition, the linear increase of memory frames with the growth of frame number also limits the ability of the models to handle long videos. To this end, we propose a Quality-aware Dynamic Memory Network (QDMN) to evaluate the segmentation quality of each frame, allowing the memory bank to selectively store accurately segmented frames to prevent the error accumulation problem. Then, we combine the segmentation quality with temporal consistency to dynamically update the memory bank to improve the practicability of the models. Without any bells and whistles, our QDMN achieves new state-of-the-art performance on both DAVIS and YouTube-VOS benchmarks. Moreover, extensive experiments demonstrate that the proposed Quality Assessment Module (QAM) can be applied to memory-based methods as generic plugins and significantly improves performance. Our source code is available at https://github.com/workforai/QDMN.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Semi-Supervised Video Object Segmentation DAVIS 2016 QDMN Jaccard (Mean) 90.7 # 8
F-measure (Mean) 93.2 # 15
J&F 92.0 # 11
Semi-Supervised Video Object Segmentation DAVIS 2017 (test-dev) QDMN J&F 81.9 # 12
Jaccard (Mean) 78.1 # 12
F-measure (Mean) 85.4 # 12
Semi-Supervised Video Object Segmentation DAVIS 2017 (val) QDMN Jaccard (Mean) 82.5 # 18
F-measure (Mean) 88.6 # 17
J&F 85.6 # 17
Semi-Supervised Video Object Segmentation YouTube-VOS 2018 QDMN F-Measure (Seen) 87.5 # 28
F-Measure (Unseen) 86.4 # 25
Overall 83.8 # 26
Jaccard (Seen) 82.7 # 27
Jaccard (Unseen) 78.4 # 19

Methods