Skip to content

Latest commit

 

History

History
25 lines (17 loc) · 1.41 KB

README.md

File metadata and controls

25 lines (17 loc) · 1.41 KB

PokeVAE

An application of VAE (Variational Auto Encoder) to encode and generate pictures of Pokemon.

The dataset contains 851 pictures of Pokemon with a size of 256x256.

The Encoder and Decoder architecture is based on this paper: https://arxiv.org/pdf/1610.00291.pdf

Beta value for KL Divergence can be set to >1 for Beta-VAE application.

Below are examples made using Beta = 0.5, LATENT_DIM = 256, and linear interpolation between the latent representation of 2 images.