Skip to content

RyanWu2233/Style_GAN2_TWDNE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 

Repository files navigation

Style_GAN2_TWDNE (This Waifu Does not exist)

Python 3.7 TensorFlow 2.10 Repo TWDNE Image size 512

Result_6

Implementation detail

The virtual Waifu pictures are generate by AI using NVIDIA famous style GAN2 algorithm. The training set is composed of 2500 images generated by TWDNE website. Resolution of each image is 512 x 512.


Reference

Style GAN2 paper: https://arxiv.org/abs/1912.04958
TWDNE website: https://www.thiswaifudoesnotexist.net/
Style GAN2 for HD fake human face: https://github.com/RyanWu2233/Style_GAN2_FFHQ
NVIDIA official source code: https://github.com/NVlabs/stylegan2


Training progress

Training


More example

Result_8a

Gif


Usage:

Execute or include file named SG2_main.pu. Then execute following instruction:

model = GAN() Create style GAN2 object
model.train(restart=False) Train model model.predict(24) Generate fake image with 4 rows and 6 cols
model.predict(6) Generate fake image with 2 rows and 3 cols
model.predict(2) Generate fake image with 1 rows and 2 cols
model.save_weights() Save model
mdoel.load_weights() Load model


About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages