Skip to content

PadminiG3404/StyloMorph

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

21 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

StyloMorph

Transform Faces into Art with Precision and Simplicity

Open in Colab Research Paper

Overview

StyloMorph is a streamlined face stylization system designed to transform facial features into artistic styles while preserving unique characteristics. Powered by advanced models like StyleGAN2 and e4e, StyloMorph ensures high-quality results with minimal computational requirements, making artistic transformations accessible to everyone.

Key Features

  • One-Shot Stylization: Transforms faces with just a single reference style image.
  • High-Quality Outputs: Maintains facial details while applying stylized transformations.
  • Efficient Processing: Uses e4e encoding for optimized GAN inversion, reducing computation time.
  • Real-Time Stylization: Works efficiently even on resource-limited hardware.
  • Customizable Style Intensity: Fine-tune outputs with adjustable parameters.
  • Supports Multi-Shot Stylization: Extends capabilities to multiple reference styles.
  • User-Friendly: Seamless integration with Google Colab for easy setup and usage.

Published Work

This research has been accepted at the 3rd DELCON, 2024 IEEE International Conference on Advancing Technology for Sustainable Development and published in IEEE.
πŸ“‘Research Paper)
Title: One-Shot Stylization for Transformative Facial Art using StyloMorph Conference: 2024 3rd Edition of IEEE Delhi Section Flagship Conference (DELCON) Publisher: IEEE


How It Works

Architecture of StloMorph

StyloMorph uses a simple yet powerful pipeline for face stylization:

  1. Facial Feature Extraction: Uses DLib Shape Predictor (68 facial landmarks) to detect key facial features.

  1. Latent Space Encoding: Maps facial features into StyleGAN2’s latent space using the e4e encoder.

  1. Style Transfer: Applies artistic transformations while preserving facial identity.

  1. Output Generation: Generates realistic, high-resolution stylized images.

πŸ”Ή Core Pre-trained Models Used:

  • StyleGAN2-FFHQ Config-F: Optimized for realistic face generation.
  • e4e-FFHQ Encode: Enables precise editing in latent space.
  • DLib 68-Landmark Shape Predictor: Facilitates landmark-based facial mapping.

Getting Started

Prerequisites

  • Python 3.8+
  • CUDA-compatible GPU (optional for faster processing)
  • Dependencies: torch, torchvision, numpy, opencv-python, dlib

Run the Notebook

Start using StyloMorph instantly by opening our pre-configured Colab notebook:

Open in Colab

Steps to Use

  1. Open the Colab notebook using the link above.
  2. Upload your source image and reference style image when prompted.
  3. Run the cells to generate your stylized output.
  4. Download the results directly from the notebook.

Notebook Features

  • Interactive Interface: Simple cell-based execution for ease of use.
  • Pre-Trained Models: Automatically downloads StyleGAN2 and e4e models for effortless setup.
  • Customizable Options: Adjust style intensity and other parameters to fine-tune results.
  • GPU Support: Leverages Google Colab's free GPU for faster processing.

Example Outputs

Source Image Style Reference Stylized Output
Source Image Style Image Stylized Output

Applications

  • 🎨 Digital Art – Create stylized portraits with unique aesthetics.
  • πŸ“Έ Content Creation – Generate artistic images for social media & branding.
  • 🎭 Entertainment & Media – Enhance visuals with creative transformations.
  • πŸ“š AI Research – Explore generative AI and style transfer techniques.

Future Scope

  • πŸš€ Real-Time Video Stylization – Extend capabilities to video processing.
  • πŸ–Œ Full-Body Stylization – Expand beyond facial transformations.
  • 🎨 AR/VR Integration – Enable immersive artistic experiences.
  • πŸ›  Interactive AI Tools – Develop user-friendly creative applications.

Acknowledgments

  • StyleGAN2 and e4e models for their foundational contributions.
  • DLib for robust facial landmark detection.
  • The FFHQ dataset for providing diverse training data.

Citation

If you use this work, please cite:

@INPROCEEDINGS{10866661,
  author={G., Uday Kiran and V., Srilakshmi and G., Padmini and B., Lavanya and A., Tejaswini and G., Vijay and D., Ranveer and B., Priyanka and H., Bhagya Laxmi},
  booktitle={2024 3rd Edition of IEEE Delhi Section Flagship Conference (DELCON)}, 
  title={One-Shot Stylization for Transformative Facial Art using StyloMorph}, 
  year={2024},
  volume={},
  number={},
  pages={1-8},
  keywords={Training;Shape;Computational modeling;Lighting;Streaming media;Media;Robustness;Real-time systems;Faces;Facial features;Face Stylization;pretrained models;Generative Adversarial Networks (GANs);Deep Learning},
  doi={10.1109/DELCON64804.2024.10866661}}

Contact

For questions or collaborations, feel free to reach out:

About

StyloMorph is an AI-powered face stylization tool using StyleGAN2 and e4e for high-quality, real-time transformations. With DLib's facial landmark detection, it ensures precision and is easily accessible via Google Colab.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages