neoql / baldgan

GAN model making people bald

BaldGAN

This repository contains:

• The source code for training the GAN model making people bald
• Data with bald people for training a neural network
• Pretrained neural network weights

Training data

Data consists of two parts:

• Bald people
• Wigs

Dataset	Image count
Bald people	3 698
Wigs	48

Bald people: google disk

Bald people preprocessed for GAN: google disk

Wigs: google disk

Bald people were obtained from:

• CelebA with manual filtration
• Collected manually from pinterest.com

Training

Pretrained model: google disk

The project is based on code from the keras-gan repository.

As a face detector, RetinaFace is used.

Project structure:

• pix2pix_InsNorm.py - network training script
• data_loader_alpha_sintes.py - data generator
• dataset_prepare.ipynb - notebook for data preparation (or just download the prepared)
• check_results_v2_InsNorm_4x4.ipynb - notebook for testing in images

To start training:

$ CUDA_VISIBLE_DEVICES=1 python3 pix2pix_InsNorm.py

To achieve the best result, it is better to pretrain the generator without a discriminator.

Features of the used model:

• Two discriminators: for the whole face and the area under the hair
• Using perceptual loss
• InstanceNormalization