HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

CTVIS: Consistent Training for Online Video Instance Segmentation

Kaining Ying Qing Zhong Weian Mao Zhenhua Wang Hao Chen Lin Yuanbo Wu Yifan Liu Chengxiang Fan Yunzhi Zhuge Chunhua Shen

CTVIS: Consistent Training for Online Video Instance Segmentation

Abstract

The discrimination of instance embeddings plays a vital role in associating instances across time for online video instance segmentation (VIS). Instance embedding learning is directly supervised by the contrastive loss computed upon the contrastive items (CIs), which are sets of anchor/positive/negative embeddings. Recent online VIS methods leverage CIs sourced from one reference frame only, which we argue is insufficient for learning highly discriminative embeddings. Intuitively, a possible strategy to enhance CIs is replicating the inference phase during training. To this end, we propose a simple yet effective training strategy, called Consistent Training for Online VIS (CTVIS), which devotes to aligning the training and inference pipelines in terms of building CIs. Specifically, CTVIS constructs CIs by referring inference the momentum-averaged embedding and the memory bank storage mechanisms, and adding noise to the relevant embeddings. Such an extension allows a reliable comparison between embeddings of current instances and the stable representations of historical instances, thereby conferring an advantage in modeling VIS challenges such as occlusion, re-identification, and deformation. Empirically, CTVIS outstrips the SOTA VIS models by up to +5.0 points on three VIS benchmarks, including YTVIS19 (55.1% AP), YTVIS21 (50.1% AP) and OVIS (35.5% AP). Furthermore, we find that pseudo-videos transformed from images can train robust models surpassing fully-supervised ones.

Code Repositories

kainingying/ctvis
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
video-instance-segmentation-on-ovis-1CTVIS (ResNet-50)
AP50: 60.8
AP75: 34.9
APho: 16.1
APmo: 41.9
mask AP: 35.5
video-instance-segmentation-on-ovis-1CTVIS (Swin-L)
AP50: 71.5
AP75: 47.5
APho: 19.1
APmo: 52.1
mask AP: 46.9
video-instance-segmentation-on-youtube-vis-3CTVIS (ResNet-50)
mAP_L: 39.4
video-instance-segmentation-on-youtube-vis-3CTVIS (Swin-L)
mAP_L: 46.4

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp