Skip to content

Latest commit

 

History

History
57 lines (35 loc) · 1.99 KB

README.md

File metadata and controls

57 lines (35 loc) · 1.99 KB

PokGAN GIF PokGAN
Generating New Pokemon Using Two-Stage Generative Adversarial Networks

Introduction

Generative models via adversarial process (GANs) are a class of machine learning models that are widely used in the field of artificial intelligence and specifically in the domain of generative modeling. Our objective is to develop a GAN-based system capable of generating novel and visually compelling Pokémon spirits that align with the patterns and aesthetics found in the provided dataset. The GAN architecture consists of a generator and a discriminator to learn the patterns, characteristics, and visual qualities exhibited by Pokémon in the custom dataset with over 1000 images. we also trained an Autoencoding GAN on the same dataset and compared the two results.


Training the GAN

#Copy all the training images in the input folder

#install the requirements
pip3 install torch torchvision

#Run the code
python3 pokegan.py

Training the AEGAN

#Copy all the images to the data/sprites_rgb folder in AEGAN

#install the requirements of requirements.txt file

#Run the code
python3 main.py

Output of GAN


Output of AEGAN Generator

Output of AEGAN Reconstructor