/Cat_WGAN

A cat face generator using WGAN.

Primary LanguagePythonMIT LicenseMIT

Cat WGAN

Generate your unique cat with one click!

The training dataset can be found here.

Preview

Usage

  1. pip install -r requirements.txt
  2. py run.py

Additional Info

This project is portable for any legal image dataset, which means you can train your own model on a given dataset.

To train a new model, extract images into dara/training/dataset, then configure pokemon_generator.config.training and pokemon_generator.config.data according to your own preferences. Additionally, don't forget to write your model name in train.py and run.py.