Virtual Hair Colour Try-On
Try different hair colors on your face or on a photo of you
The Virtual Hair Colour Try-On System for Hair Makeovers is a deep learning based application that allows users to experiment with different hair colors on their photos. By leveraging a U-Net architecture for image segmentation, this system precisely isolates the hair region and applies various hair colors, enabling a realistic makeover effect.
Click to import image
Output image
Technologies Used
Deep Learning: PyTorch, U-Net (for hair segmentation)
Model Conversion: Onnx (for model optimization and deployment)
Web Framework: Flask (for building the web app)
Deployment: Docker, GCP (for containerization and scalability)
Project repository