Skip to content

rexnxiaobai/Stacked-GAN-Face-SR

 
 

Repository files navigation

Stacked-GAN-Face-SR

Face Synthesis and Super Resolution Using Stacked Generative Adversarial Network

1. Dataset generate

  • a.Using HD CelebA Cropper to generate the dataset raw data.
  • b.Use the default 0.7 face_factor.
  • c.Copy the raw data to dataset folder.
  • c.Run 1.get_celeb_train.py to generate training, validating and testing data

2. Training

  • a.Run the training command python 2.train_and_test.py --phase train --dataset_name yourData --origin_size 64 --image_size 256
  • b.Check the sample folder during training to make sure the result is correct.

3. Testing

  • a.Run the testing command python 2.train_and_test.py --phase test --dataset_name yourData --origin_size 64 --image_size 256
  • b.Check the generated test folder for the results

4. Calculate FID

  • a.The FID calculating code is from here
  • b.Run 3.split_testset.py to seperate blur tesing images and ground truth
  • c.Run the FID calculating command python 4.fid_score.py dataset/yourData/gt test

5. FID results

Table 1 FID score compare in different resolution transformation

name 16 to 64 32 to 128 64 to 256
Wavelet SRNet 45.952 54.533 31.929
Cycle GAN 63.837 32.228 42.218
SRN-Deblur 125.576 41.463 12.729
ours 51.646 29.685 13.431

6. Images results

Fig

About

Stacked GAN face super resolution

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%