Dense Unsupervised Learning for Video Segmentation

Araslanov, Nikita; Schaub-Meyer, Simone; Roth, Stefan

Open Access

Dense Unsupervised Learning for Video Segmentation

Files

dense-ulearn-vos-code.zip (Training and inference code (PyTorch)) (7.65 MB)

snapshots.tar.gz (Model parameters (snapshots)) (402.41 MB)

results.tar.gz (Inference results) (8.38 GB)

Date

2023-08-04

Type

Software

Authors

Araslanov, Nikita
Schaub-Meyer, Simone

Roth, Stefan

Description

We present a novel approach to unsupervised learning for video object segmentation (VOS). Unlike previous work, our formulation allows to learn dense feature representations directly in a fully convolutional regime. We rely on uniform grid sampling to extract a set of anchors and train our model to disambiguate between them on both inter- and intra-video levels. However, a naive scheme to train such a model results in a degenerate solution. We propose to prevent this with a simple regularisation scheme, accommodating the equivariance property of the segmentation task to similarity transformations. Our training objective admits efficient implementation and exhibits fast training convergence. On established VOS benchmarks, our approach exceeds the segmentation accuracy of previous work despite using significantly less training data and compute power.