Skip to content

Latest commit

 

History

History
 
 

Autoencoder-in-TensorFlow

Introduction to Autoencoder in TensorFlow

This repsitory contains code for Autoencoder in TensorFlow 2: Beginner’s Guide blogpost.

Package Dependencies

download

This repository also trains the Autoencoder in Tensorflow framework on Fashion-MNIST and Cartoon dataset. The cartoon dataset can be downloaded from here.

The code is tested with:

  • Cuda-11.1
  • Cudnn-8.0

The Tensorflow notebook requires numpy, tf-nightly-gpu, opencv, sklearn. To get the versions of these packages you need for the program, use pip: (Make sure pip is upgraded: python3 -m pip install -U pip)

pip3 install -r requirements.txt 

Add Virtualenv as Python Kernel in Jupyterlab

  • Activate the virtualenv
$ source your-venv/bin/activate
  • Add the virtualenv as a jupyter kernel
(your-venv)$ ipython kernel install --name "local-venv" --user

Replace local-venv with your virtualenv name.

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.

img