Disentangled Pre-training for Image Matting

Yanda Li, Zilong Huang, Gang Yu, Ling Chen, Yunchao Wei +1 more
4/3/2023
cs.CV

No official code found. Best implementation: Yanda95/dpt (6 stars, Apache-2.0, last commit 2023-12-06).

Abstract

Image matting requires high-quality pixel-level human annotations to support the training of a deep model in recent literature. Whereas such annotation is costly and hard to scale, significantly holding back the development of the research. In this work, we make the first attempt towards addressing this problem, by proposing a self-supervised pre-training approach that can leverage infinite numbers of data to boost the matting performance. The pre-training task is designed in a similar manner as image matting, where random trimap and alpha matte are generated to achieve an image disentanglement objective. The pre-trained model is then used as an initialisation of the downstream matting task for fine-tuning. Extensive experimental evaluations show that the proposed approach outperforms both the state-of-the-art matting methods and other alternative self-supervised initialisation approaches by a large margin. We also show the robustness of the proposed approach over different backbone architectures. Our project page is available at https://crystraldo.github.io/dpt_mat/.

View on arXivView PDF

Code Implementations(1)

Yanda95/dptCommunity100%
Apache-2.0
61PythonNov 17, 20222 years agoApache-2.0

Similar Papers

Cite this paper

@article{li2023disentangled,
  title  = {Disentangled Pre-training for Image Matting},
  author = {Yanda Li and Zilong Huang and Gang Yu and Ling Chen and Yunchao Wei and Jianbo Jiao},
  year   = {2023},
  eprint = {2304.00784},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2304.00784v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2304.00784.svg)](https://codeofpaper.com/paper/2304.00784)

Discussion