🌟 Aarush Singh Kushwaha 🎓
🌟 Hardik Tiwari 🎓
🌟 Ikrima Badr Shamim Ahmed 🎓
🌟 Meher Narula 🎓
--> Link to HuggingFace🤗 Deployment
-->Link to weights
--> This Notebook contains our code.
NOTE: You would have to change paths to run the notebook entirely.
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.
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.
- 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.
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
The UTK Face dataset, which includes a wide range of ages and demographics, is used for training and validating the model.
We tried the model on some of our own images, hope you would like the effort 😄🎉