IronDressAI: AI-Powered Wrinkle Removal Application

Shivansh Yadav

ML Engineer
Software Engineer
AI Developer
Hugging Face
Python
Stable Diffusion

IronDressAI 👕: AI-Powered Wrinkle Removal Application

An innovative AI application that automatically removes wrinkles from clothing in images using fine-tuned SDXL-Turbo model, featuring both automatic and manual masking capabilities for precise wrinkle detection and removal.

Project Description:

IronDressAI is an application that leverages the power of Stable Diffusion XL Turbo to digitally "iron" wrinkled clothing in photographs. Through parameter-efficient fine-tuning using LoRA on a curated dataset, we developed a specialized model that produces high-quality, wrinkle-free results while maintaining the original garment's texture and design.
Technologies Used:
SDXL-Turbo (Base Model)
Kohya LoRA (Model Training)
MediaPipe (Automatic Masking)
NextJS (Frontend)
Replicate (Model Deployment)
Cog (Docker Containerization)
HuggingFace (Dataset Management)
The application features both automatic masking using MediaPipe and an intuitive manual masking interface, allowing users to precisely select wrinkled areas for processing. Deployed as a containerized application through Replicate, it offers seamless API integration and real-time processing capabilities.
IronDressAI was a team project and is open-source: https://github.com/Team-Iron-The-Dress/WebUI_for_Iron_the_dress_custom

Looking to build something similar?

As a no-code developer with a strong coding background, I bring the best of both worlds to your project.
Whether you're interested in AI-powered applications, automated workflows, or custom business solutions, I can help bring your vision to life efficiently and cost-effectively.
Let's connect and discuss how we can transform your ideas into reality!
Partner With Shivansh
View Services

More Projects by Shivansh