TfVaeMNIST

Reference

https://www.tensorflow.org/tutorials/generative/cvae

Files

• tfcnn.py
  • Main file of the Variational Autoencoder
  • Train by excluding one of the numbers 0 to 9
  • Test every number 0 to 9
• tfgpu.py
  • Initialize the GPU devices
• tfmnist.py
  • Load the MNIST dataset
  • Normalize images

Development Environment

• Python 3.7.5
• Tensorflow 2.1
• IntelliJ IDEA 2019.3.3 (Community Edition)
  • Plugin: Python Community Edition 193.6494.42