Skip to content

AaSiKu/Assignment-4-IllutionCraft

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

Illusion Craft 🎨✨

Summer Project — Programming Club 🌞💻

Members:

🌟 Aarush Singh Kushwaha 🎓
🌟 Hardik Tiwari 🎓
🌟 Ikrima Badr Shamim Ahmed 🎓
🌟 Meher Narula 🎓

Important Links:

--> Link to HuggingFace🤗 Deployment
-->Link to weights
--> This Notebook contains our code.

NOTE: You would have to change paths to run the notebook entirely.

Contents 📑

  1. Problem Statement
  2. Cycle-GAN
  3. Dataset

Problem Statement 📋

Develop and deploy a CycleGAN-based model capable of aging and de-aging facial images. The deployment includes creating an interactive web application using Streamlit and hosting the model on Hugging Face for scalability. The UTK Face dataset is used for training and validation.

Cycle-GAN 🔄🎨

CycleGAN enables image-to-image translation, even with unpaired datasets, by treating it as an image reconstruction problem. It uses two generators (G and F) and two discriminators (Dx and Dy) to perform forward and backward cycles, ensuring consistency and realistic outputs.

Architecture 🏛️

  • Generators:
    • G: Converts input image ( x ) to the aged/de-aged image.
    • F: Converts the aged/de-aged image back to the original image.
  • Discriminators: Distinguish between real and generated images.
  • Loss Functions:
    • Adversarial Loss: Ensures generated images are realistic.
    • Cycle Consistency Loss: Ensures the transformations are accurate by comparing the input and reconstructed images.

Deployment 🚀🌐

The model is deployed using:

  • Streamlit: For creating an interactive web application.
  • Hugging Face: For hosting the model to ensure ease of access and scalability.
    Hugging Face link

Dataset 📊

The UTK Face dataset, which includes a wide range of ages and demographics, is used for training and validating the model.


Results 🏆

We tried the model on some of our own images, hope you would like the effort 😄🎉

Result 1

Result 1

Result 1

Result 1

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published