This repository contains the official PyTorch implementation of the paper Region-Aware Face Swapping (CVPR2022).
conda env create -f environment/psp_env.yaml
Download the CelebA-HQ to the path Data and create mask annotations.
cd scripts
python mask_npy.py
Download pre-trained StyleGAN, IR-SE50, and StyleGAN inversion models to the path pretrained_models, which could be downloaded following the instructions of original pSp Github repository.
python scripts/train.py
Download pre-trained model swap to the path checkpoints
python inference_celeba.py