This repsitory contains code for Autoencoder in TensorFlow 2: Beginner’s Guide blogpost.
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
- 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.
Want to become an expert in AI? AI Courses by OpenCV is a great place to start.