Skip to content

stylegan2, tensorflow 2, keras subclassing

Notifications You must be signed in to change notification settings

nyroro/stylegan2-tf-2.x

 
 

Repository files navigation

stylegan2-tf-2.x

  • This is tensorflow 2.0 based keras subclassing reimplementation of official StyleGAN2 Repo

To do list

Items Implemented
weight modulation / demodulation ✔️
skip architecture ✔️
resnet architecture ✔️
Path regularization ✔️
Lazy regularization ✔️
Fast optimized bias / activation / bilinear filtering (cuda compiled code) -
Single GPU training ✔️
Multi GPU distributed training -
Inference from official Generator weights ✔️

Inference from official weights

Official repo result Official weight copied result from this repo
official_result restored_result

Current result

at 284k train step Screenshot
Loss loss_tensorboard
Generation output

Real Images

phi=0.0

phi=0.5

phi=0.7

phi=1.0
generation_tensorboard

About

stylegan2, tensorflow 2, keras subclassing

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%