Touch-Up-The-Hair

Image Processing Workflow

This section describes the workflow for processing images using our application.

graph TD;
    style Upload_Image fill:#64B5F6,stroke:#333,stroke-width:2px,stroke-dasharray: 5, 5,font-size:14px,font-weight:normal,font-family:Arial;
    style Generate_Mask fill:#81C784,stroke:#333,stroke-width:2px,stroke-dasharray: 5, 5,font-size:14px,font-weight:normal,font-family:Arial;
    style Inpainting fill:#FFD54F,stroke:#333,stroke-width:2px,stroke-dasharray: 5, 5,font-size:14px,font-weight:normal,font-family:Arial;
    style Touch_up_Image fill:#FF8A65,stroke:#333,stroke-width:2px,stroke-dasharray: 5, 5,font-size:14px,font-weight:normal,font-family:Arial;
    style Download_Image fill:#A1887F,stroke:#333,stroke-width:2px,stroke-dasharray: 5, 5,font-size:14px,font-weight:normal,font-family:Arial;

    Upload_Image["Upload Image"] --> Generate_Mask["Generate Mask"];
    Generate_Mask --> Inpainting["Inpainting using stable diffusion with ControlNet"];
    Inpainting --> Touch_up_Image["Touch up Image"];
    Touch_up_Image --> Download_Image["Download Image"];

Loading

Objective

The objective of this project is to touch up hair in the image to match the minority hair color to the previously applied predominant color, resulting in a beautiful, uniform single-color hair appearance.

Project Overview

This project employs stable diffusion inpainting with ControlNet to achieve the desired hair color touch-up. The algorithm utilizes a minority hair color mask to guide the inpainting process.

Usage

1. Clone this repository to Google Colab.
2. Open and run the notebook touch_up_the_hair.ipynb in Google Colab.

Repository Structure

• touch_up_the_hair.ipynb: Contains the Python code for the hair color touch-up process.
• input_dir/: Directory to upload input images.
• output_dir/: Directory where the processed images will be saved.
• models/: Directory to store required models.
• Images/: Directory containing test images

Models

• selfie_multiclass_256x256.tflite: Mediapipe model for segmenting hair in images.

Note

If you intend to use images other than the provided test images, please ensure that your IMG_PATH variable is up-to-date. We recommend utilizing the pre-built input_dir directory, where you can conveniently upload your images after creating the directory.