Skip to content

Latest commit

 

History

History
71 lines (58 loc) · 3.12 KB

README.md

File metadata and controls

71 lines (58 loc) · 3.12 KB

Style Transfer - Analytics Club, IIT Madras


Teaser Video: Top row denotes source images, and the leftmost column denotes the reference images from which the style is being transferred.

Repository Contents

Links

  • celeb_a dataset (male, female) - full, sample
  • celeb_a attribute dataset - link
  • pretrained models - starganv2, classifier
  • Session 1 recording - Intro to GANs, Understanding the paper.
  • Session 2 recording part 1 - Dataloaders and basic model blocks, part 2 - complete model, training and validation helper functions.
  • Session 3 recording - training large models on VMs, possible improvements, deploying app using dash and flask, interesting intuitions and manipulating latent space.

Interpolation with reference images

Interpolation of latent space to manipulate certain attributes

Masks generated using masking model to retain features like eyes, nose and mouth of the source image

Interesting observations

Demon-like images during cross domain interpolation

Retention of domain characteristics irrespective of reference image

Well seperated domain latent spaces

Grad Cam to visualise activations to understand each feature of the latent space