Skip to content

Lornatang/TensorFlow-WGAN_GP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

72 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TensorFlow-WGAN

Model: "Generator model"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
conv2d (Conv2D)              (None, 16, 16, 64)        1792      
_________________________________________________________________
conv2d_1 (Conv2D)            (None, 8, 8, 128)         73856     
_________________________________________________________________
flatten (Flatten)            (None, 8192)              0         
_________________________________________________________________
dense (Dense)                (None, 1)                 8193      
=================================================================
Total params: 83,841
Trainable params: 83,841
Non-trainable params: 0
_________________________________________________________________
Model: "Discriminator model"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
dense_1 (Dense)              (None, 8192)              1048576   
_________________________________________________________________
reshape (Reshape)            (None, 8, 8, 128)         0         
_________________________________________________________________
conv2d_transpose (Conv2DTran (None, 16, 16, 128)       147456    
_________________________________________________________________
conv2d_transpose_1 (Conv2DTr (None, 32, 32, 64)        73728     
_________________________________________________________________
conv2d_transpose_2 (Conv2DTr (None, 32, 32, 3)         1728      
=================================================================
Total params: 1,271,488
Trainable params: 1,271,488
Non-trainable params: 0
_________________________________________________________________

About

An implements for Wasserstein GAN by TensorFlow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages