We aim to remove facial masks from images using a novel neural network architecture. Facial masks have been, and will be part of many people's lives. UnVeilify can remove these face masks from any image, given a masked image and an identity image, while preserving the identity features.
Documentation of our project can be seen in our documentation.
Update the options file and run the following command.
python train.py -c options/UnVeilify.yamlThe pretrained model required for training can be found in eladrich/pixel2style2pixel, where we use the pretrained IR-SE50 model.
Change the directory to the model checkpoint and dataset, then run the following command.
python test.py -c CHECKPOINT_DIR -d DATASET_DIR