Skip to content

Instantly share code, notes, and snippets.

@AFAgarap

AFAgarap/mini_vgg_ae.py

Created November 19, 2019 16:01
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save AFAgarap/18e720eee16626517671d6f9fce55e52 to your computer and use it in GitHub Desktop.
Save AFAgarap/18e720eee16626517671d6f9fce55e52 to your computer and use it in GitHub Desktop.
Download ZIP
TensorFlow 2.0 implementation of mini VGG-based Autoencoder.
Raw
mini_vgg_ae.py
class Autoencoder(tf.keras.Model):
def __init__(self, **kwargs):
super(Autoencoder, self).__init__()
self.encoder = Encoder(
input_shape=kwargs['input_shape'],
latent_dim=kwargs['latent_dim']
)
self.decoder = Decoder(latent_dim=kwargs['latent_dim'])
def call(self, features):
code = self.encoder(features)
reconstructed = self.decoder(code)
return reconstructed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment