|
2 anni fa | |
---|---|---|
.. | ||
VAE_Cartoon_TensorFlow-checkpoint.ipynb | 3 anni fa | |
VAE_FashionMnist_TensorFlow-checkpoint.ipynb | 3 anni fa | |
readme-checkpoint.md | 2 anni fa | |
requirements-checkpoint.txt | 3 anni fa |
This repository trains the Variational Autoencoder in Tensorflow framework on Fashion-MNIST and Cartoon dataset. The cartoon dataset can be download 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
$ source your-venv/bin/activate
(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.