/Variational-Auto-Encoders

A small implementation of variational autoencoder setup, in convolutional neural networks. I tried to learn generative modeling, and in that used this setup to train a neural network to restore given images.

Primary LanguagePython

"# Variational-Auto-Encoders"

Steps to use this network:

  1. Create a folder named Datasets.
  2. Keep all your training images in Datasets/Dataset_Folder_Name/train_images
  3. Just run VAE.py to train the network.