/AI_TryOn_mini

An AI try-on application for generating photos with AI character wearing the same clothes as the one in the input photo.

Primary LanguagePython

AI_TryOn_mini

A lightweight AI-driven try-on web app that visualizes an AI character donning your dream outfit. Upload, segment, render, all is seamless and effortless.

Please note: All images utilized herein were sourced from publicly accessible online databases and are used for illustrative purposes only.

Demo

Tools

From top to bottom

  • Clear image
  • Drawer
  • SAM point-segmenter with interactive functionality (left pos right neg)
  • Rect-drawer for local area correction
  • Undo
  • Eraser
  • Expand
  • Download
  • Send to render

Colab Tutorial

Follow this Colab example, or run on Colab. Need to register a ngrok account and copy your token to replace "{your_token}".

Run Locally

  • Install the dependencies (if needed)
pip install -r requirements.txt

You may download models from civitai or

./script/download.sh

or

. script/download.sh

where the model I downloaded from official sites and distributors on civitai. Thanks to them.

  • Launch backend
python run_app.py
  • Go to Browser
http://127.0.0.1:8001

Hardware requirements

  • At least 12G GPU memory is required

TODO

  • CodeFormer
  • Memory optimization
  • ...