This is the official repository of the paper [IJCV'23] Rethinking Portrait Matting with Privacy Preserving.
For further questions, please contact Sihan Ma at sima7436@uni.sydney.edu.au or Jizhizi Li at jili8515@uni.sydney.edu.au.
Introduction | PPT and P3M-10k | P3M-Net | P3M-CP | Results | Train | Inference code | Statement
[2024-3-31]: Code for training is available!
[2023-11-05]: Publish the ViTAE-S and SWIN-T backbone models pretrained on ImageNet that can be used to train our P3M-Net from scratch.
[2023-03-28]: The paper has been accepted by the International Journal of Computer Vision (IJCV)! 🎉
[2022-03-31]: Publish the inference code and the pretrained model that can be used to test with our SOTA model P3M-Net(ViTAE-S) on your own privacy-preserving or normal portrait images.
[2021-12-06]: Publish the P3M-10k dataset.
[2021-11-21]: Publish the conference paper ACM MM 2021 "Privacy-preserving Portrait Matting". The code and data are available at github repo.
Other applications of ViTAE Transformer include: image classification | object detection | semantic segmentation | animal pose segmentation | remote sensing
Recently, there has been an increasing concern about the privacy issue raised by using personally identifiable information in machine learning. However, previous portrait matting methods were all based on identifiable portrait images.
To fill the gap, we present P3M-10k in this paper, which is the first large-scale anonymized benchmark for Privacy-Preserving Portrait Matting. P3M-10k consists of 10,000 high-resolution face-blurred portrait images along with high-quality alpha mattes. We systematically evaluate both trimap-free and trimap-based matting methods on P3M-10k and find that existing matting methods show different generalization capabilities when following the Privacy-Preserving Training (PPT) setting, 𝑖.𝑒., training on face-blurred images and testing on arbitrary images.
To devise a better trimap-free portrait matting model, we propose P3M-Net, consisting of three carefully designed integration modules that can perform privacy-insensitive semantic perception and detail-reserved matting simultaneously. We further design multiple variants of P3MNet with different CNN and transformer backbones and identify the difference of their generalization abilities.
To further mitigate this issue, we devise a simple yet effective Copy and Paste strategy (P3M-CP) that can borrow facial information from public celebrity images without privacy concerns and direct the network to reacquire the face context at both data and feature level. P3M-CP only brings a few additional computations during training, while enabling the matting model to process both face-blurred and normal images without extra effort during inference.
Extensive experiments on P3M-10k demonstrate the superiority of P3M-Net over state-of-the-art methods and the effectiveness of P3MCP in improving the generalization ability of P3M-Net, implying a great significance of P3M for future research and real-world applications.
PPT Setting: Due to the privacy concern, we propose the Privacy-Preserving Training (PPT) setting in portrait matting, 𝑖.𝑒., training on privacy-preserved images (𝑒.𝑔., processed by face obfuscation) and testing on arbitraty images with or without privacy content. As an initial step towards privacy-preserving portrait matting problem, we only define the identifiable faces in frontal and some profile portrait images as the private content in this work.
P3M-10k Dataset: To further explore the effect of PPT setting, we establish the first large-scale privacy-preserving portrait matting benchmark named P3M-10k. It contains 10,000 annonymized high-resolution portrait images by face obfuscation along with high-quality ground truth alpha mattes. Specifically, we carefully collect, filter, and annotate about 10,000 high-resolution images from the Internet with free use license. There are 9,421 images in the training set and 500 images in the test set, denoted as P3M-500-P. In addition, we also collect and annotate another 500 public celebrity images from the Internet without face obfuscation, to evaluate the performance of matting models under the PPT setting on normal portrait images, denoted as P3M-500-NP. We show some examples as below, where (a) is from the training set, (b) is from P3M-500-P, and (c) is from P3M-500-NP.
P3M-10k and the facemask are now published!! You can get access to it from the following links, please make sure that you have read and agreed to the agreement. Note that the facemask is not used in our work. So it's optional to download it.
| Dataset | Dataset Link |
Dataset Link |
Dataset Release Agreement |
|---|---|---|---|
| P3M-10k | Link | Link (pw: cied) | Agreement (MIT License) |
Our P3M-Net network models the comprehensive interactions between the sharing encoder and two decoders through three carefully designed integration modules, i.e., 1) a tripartite-feature integration (TFI) module to enable the interaction between encoder and two decoders; 2) a deep bipartite-feature integration (dBFI) module to enhance the interaction between the encoder and segmentation decoder; and 3) a shallow bipartitefeature integration (sBFI) module to promote the interaction between the encoder and matting decoder.
We design three variants of P3M Basic Blocks based on CNN and vision transformers, namely P3M-Net(ResNet-34), P3M-Net(Swin-T), P3M-Net(ViTAE-S). We leverage the ability of transformers in modeling long-range dependency to extract more accurate global information and the locality modelling ability to reserve lots of details in the transition areas. The structures are shown in the following figures.
To further improve the generalization ability of P3M-Net, we devise a simple yet effective Copy and Paste strategy (P3M-CP) that can borrow facial information from publicly available celebrity images without privacy concerns and guide the network to reacquire the face context at both data and feature level, namely P3M-ICP and P3M-FCP. The pipeline is shown in the following figure.
We test our network on our proposed P3M-500-P and P3M-500-NP and compare with previous SOTA methods, we list the results as below.
Please follow this instruction page.
Here we provide the procedure of testing on sample images by our pretrained P3M-Net(ViTAE-S) model:
-
Setup environment following this instruction page;
-
Insert the path
REPOSITORY_ROOT_PATHin the filecore/config.py; -
Download the pretrained P3M-Net(ViTAE-S) model from here (Google Drive | Baidu Wangpan (pw: hxxy))) and unzip to the folder
models/pretrained/; -
Save your sample images in folder
samples/original/.; -
Setup parameters in the file
scripts/test_samples.shand run by:chmod +x scripts/test_samples.shscripts/test_samples.sh; -
The results of alpha matte and transparent color image will be saved in folder
samples/result_alpha/.andsamples/result_color/..
We show some sample images, the predicted alpha mattes, and their transparent results as below. We use the pretrained P3M-Net(ViTAE-S) model from section P3M-Net and Variants with `RESIZE` test strategy.
If you are interested in our work, please consider citing the following:
@article{rethink_p3m,
title={Rethinking Portrait Matting with Pirvacy Preserving},
author={Ma, Sihan and Li, Jizhizi and Zhang, Jing and Zhang, He and Tao, Dacheng},
journal={International Journal of Computer Vision},
publisher={Springer},
ISSN={1573-1405},
year={2023}
}
This project is under MIT licence.
For further questions, please contact Sihan Ma at sima7436@uni.sydney.edu.au or Jizhizi Li at jili8515@uni.sydney.edu.au.
[1] Deep Automatic Natural Image Matting, IJCAI, 2021 | Paper | Github
Jizhizi Li, Jing Zhang, and Dacheng Tao
[2] Privacy-preserving Portrait Matting, ACM MM, 2021 | Paper | Github
Jizhizi Li∗, Sihan Ma∗, Jing Zhang, Dacheng Tao
[3] Bridging Composite and Real: Towards End-to-end Deep Image Matting, IJCV, 2022 | Paper | Github
Jizhizi Li∗, Jing Zhang∗, Stephen J. Maybank, Dacheng Tao
[4] Referring Image Matting, CVPR, 2023 | Paper | Github
Jizhizi Li, Jing Zhang, and Dacheng Tao
[5] Deep Image Matting: A Comprehensive Survey, ArXiv, 2023 | Paper | Github
Jizhizi Li, Jing Zhang, and Dacheng Tao