Weakly Supervised Instance Segmentation for Videos with Temporal Mask Consistency

Abstract

Weakly supervised instance segmentation reduces the cost of annotations required to train models. However, existing approaches which rely only on image-level class labels predominantly suffer from errors due to (a) partial segmentation of objects and (b) missing object predictions. We show that these issues can be better addressed by training with weakly labeled videos instead of images. In videos, motion and temporal consistency of predictions across frames provide complementary signals which can help segmentation. We are the first to explore the use of these video signals to tackle weakly supervised instance segmentation. We propose two ways to leverage this information in our model. First, we adapt inter-pixel relation network (IRN) to effectively incorporate motion information during training. Second, we introduce a new MaskConsist module, which addresses the problem of missing object instances by transferring stable predictions between neighboring frames during training. We demonstrate that both approaches together improve the instance segmentation metric $A{P}_{50}$ on video frames of two datasets: Youtube-VIS and Cityscapes by $5\%$ and $3\%$ respectively.

arXiv comment14 pages, 8 figures, accepted by CVPR 2021

References in corpus (5)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Multiscale Combinatorial Grouping for Image Segmentation and Object Proposal Generation
• STEm-Seg: Spatio-temporal Embeddings for Instance Segmentation in Videos
• Learning a Spatio-Temporal Embedding for Video Instance Segmentation
• Label-PEnet: Sequential Label Propagation and Enhancement Networks for Weakly Supervised Instance Segmentation