sveatlo / unmasked Public

Notifications You must be signed in to change notification settings
Fork 0
Star 4

Removes face masks from faces using ML inpainting

Unlicense license

4 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 12 Commits
network		network
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
dataset.py		dataset.py
environment.yml		environment.yml
train.py		train.py
unmasked.py		unmasked.py

Repository files navigation

UnMasked

Unmasking peoples' faces using image inpainting.

github: github.com/sveatlo/unmasked

Training

Download dataset images to ./dataset/celeba/images
python train.py --gpus 1 --batch_size 16

Download pretrained model

You can download the pretrained model from Google drive

About

Removes face masks from faces using ML inpainting

pytorch gan inpainting pytorch-lightning

Unlicense license

Report repository

Languages

Python 100.0%