Skip to content

IShengFang/SpectralNormalizationKeras

Repository files navigation

Spectral Normalization for Keras

The simple Keras implementation of ICLR 2018 paper, Spectral Normalization for Generative Adversarial Networks. [openreview][arixiv][original code(chainer)]

[Hackmd][github]

Result

CIFAR10

DCGAN architecture

10epoch With SN Without SN
With GP Image Image
Without GP Image Image
100epoch With SN Without SN
With GP Image Image
Without GP Image Image
200epoch With SN Without SN
With GP Image Image
Without GP Image Image
300epoch With SN Without SN
With GP Image Image
Without GP Image Image
400epoch With SN Without SN
With GP Image Image
Without GP Image Image
500epoch with SN without SN
With GP Image Image
Without GP Image Image
Loss with SN without SN
With GP Image Image
Without GP Image Image

ResNet architecture

10epoch With SN Without SN
With GP Image Image
Without GP Image Image
100epoch With SN Without SN
With GP Image Image
Without GP Image Image
200epoch With SN Without SN
With GP Image Image
Without GP Image Image
300epoch With SN Without SN
With GP Image Image
Without GP Image Image
400epoch With SN Without SN
With GP Image Image
Without GP Image Image
500epoch with SN without SN
With GP Image Image
Without GP Image Image
Loss with SN without SN
With GP Image Image
Without GP Image Image

How to use?

  1. Move SpectralNormalizationKeras.py in your dir
  2. Import these layer class
from SpectralNormalizationKeras import DenseSN, ConvSN1D, ConvSN2D, ConvSN3D
  1. Use these layers in your discriminator as usual

Example notebook

CIFAR10 with DCGAN architecture

CIFAR10 with ResNet architecture

Model Detail

Architecture

DCGAN

Generator

Image

Discriminator

Image

ResNet GAN

Generator

Image

Generator UpSampling ResBlock

Image

Dicriminator

Image

Discriminator DownSampling ResBlock

Image

Discriminator ResBlock

Image

Issue

  • Compare with WGAN-GP
  • Projection Discriminator

Acknowledgment

  • Thank @anshkapil pointed out and @IFeelBloated corrected this implementation.

Releases

No releases published

Packages

 
 
 

Contributors