Generate your unique cat with one click!
The training dataset can be found here.
pip install -r requirements.txt
py run.py
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
.