Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Adversarially-Guided Portrait Matting

About

We present a method for generating alpha mattes using a limited data source. We pretrain a novel transformerbased model (StyleMatte) on portrait datasets. We utilize this model to provide image-mask pairs for the StyleGAN3-based network (StyleMatteGAN). This network is trained unsupervisedly and generates previously unseen imagemask training pairs that are fed back to StyleMatte. We demonstrate that the performance of the matte pulling network improves during this cycle and obtains top results on the human portraits and state-of-the-art metrics on animals dataset. Furthermore, StyleMatteGAN provides high-resolution, privacy-preserving portraits with alpha mattes, making it suitable for various image composition tasks. Our code is available at https://github.com/chroneus/stylematte

Sergej Chicherin, Karen Efremyan• 2023

Related benchmarks

TaskDatasetResultRank
Image GenerationFFHQ
FID5.13
52
Image MattingP3M-500-NP
SAD (Trimap)6.639
27
Matte PullingP3M-500-P face-blurred
SAD5.85
11
Image GenerationAFHQ v2
FID4.23
11
Image MattingAM-2k (test)
SAD9.602
9
Showing 5 of 5 rows

Other info

Code

Follow for update