Xunius / CycleGAN_emoji

Swap Apple-Windows styles emojis using CycleGAN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CycleGAN_emoji

Swap Apple-Windows styles emojis using CycleGAN

This is based on an assignment of CSC321-Winter-2018, University of Toronto.

The task

In this assignment, you’ll get hands-on experience coding and training GANs. This assignment is divided into two parts: in the first part, we will implement a specific type of GAN designed to process images, called a Deep Convolutional GAN (DCGAN). We’ll train the DCGAN to generate emojis from samples of random noise. In the second part, we will implement a more complex GAN architecture called CycleGAN, which was designed for the task of image-to-image translation (described in more detail in Part 2). We’ll train the CycleGAN to convert between Apple-style and Windows-style emojis.

Dependencies

  • PyTorch (developed in 1.11.0)
  • torchvision (developed in 0.12.0)

Inventory

  • config.py: configuration settings for training.
  • model.py: model definitions.
  • loader.py: data loader.
  • train.py: training script.
  • test.py: perform inference using trained model.

Results

The following are some random inference results on the test dataset, using pre-trained weights trained after 100 epochs.

The first row contains Apple-to-Windows style-transferred emojis, and the 2nd row the reverse transfers.

About

Swap Apple-Windows styles emojis using CycleGAN

License:MIT License


Languages

Language:Python 100.0%