This repository includes the official project of Mask Guided (MG) Matting, presented in our paper:
Mask Guided Matting via Progressive Refinement Network
Johns Hopkins University, Adobe Research
- Trimap-free Alpha Estimation: MG Matting does not require a carefully annotated trimap as guidance inputs. Instead, it takes a general rough mask, which could be generated by segmentation or saliency models automatically, and predicts an alpha matte with great details;
- Foreground Color Prediction: MG Matting predicts the foreground color besides alpha matte, we notice and address the inaccuracy of foreground annotations in Composition-1k by Random Alpha Blending;
- No Additional Training Data: MG Matting is trained only with the widely-used publicly avaliable synthetic dataset Composition-1k, and shows great performance on both synthetic and real-world benchmarks.
-
15 Dec 2020: Visually comparisons of different fully automatic matting systems are avaliable in SYSTEM.md.
-
15 Dec 2020: Release Arxiv version of paper and visualizations of sample images and videos.
- Inference demo, real-world portrait benchmark shall be released soon. Before that, if you want to test your model on the real-world portrait benchmark or compare results with MG Matting, feel free to contact Qihang Yu (yucornetto@gmail.com).
We provide examples for visually comparing MG Matting with other matting methods. We also note that our model can even potentially deal with video matting.
In addition, we also use an internal portrait dataset consisting of 4395 images to train an even stronger MG Matting model, and construct a fully automatic matting system based on it. We provide visually comparison of this automatic matting system with other latest matting methods including MODNet and commercial softwares such as Adobe PhotoShop.
Please refer to RESULT.md for more visualization results.
In our experiments, only Composition-1k training set is used to train the model. And the obtained model is evaluated on three dataset: Composition-1k, Distinction-646, and our real-world portrait dataset.
For Compsition-1k, please contact Brian Price (bprice@adobe.com) requesting for the dataset. And please refer to GCA Matting for dataset preparation.
For Distinction-646, please refer to HAttMatting for the dataset.
Our real-world portrait dataset shall be released to public soon.
If you find this work or code useful for your research, please use the following BibTex entry:
@article{yu2020mask,
author = {Yu, Qihang and Zhang, Jianming and Zhang, He and Wang, Yilin and Lin, Zhe and Xu, Ning and Bai, Yutong and Yuille, Alan},
title = {Mask Guided Matting via Progressive Refinement Network},
journal = {arXiv preprint arXiv:2012.06722},
year = {2020}
}
Research only