xanjay / cmate-virtual-tryon

Virtual clothing try on tool powered by deep learning.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool



About CMate

CMate is a virtual clothing try on tool. Given a profile image and a source image containing cloth, it applies the cloth to your profile image and let you see how does it look on you as if you were actually wearing it.
It is a flask web application powered by deep learning. Demo video: https://youtu.be/M4NjXy27Yrs.


Features

  • Supports all kinds of upper wear clothes for both men and women.
  • Accurate cloth extraction and fitting.
  • Simple, elegent and easy to use web app.

How does it work ?

It works by extractng cloth from source image and fitting into your profile image.
Under the hood, it uses two separate deep learning models:
Cloth segmentation model: Custom Deeplab model to extract cloth from source image.
Pose estimator model: Pretrained Openpose body_25 model used to locate shoulder points.
Extracted cloth is blended into profile image based on shoulder location.

Dataset

  • Deepfashion2 Dataset: used to train cloth segmentation model.

Installation

Use docker image OR see installation.md.

References

Contributing

If you have any issue fixes or improvement changes. Fork this repo, make changes and submit pull request.

License

CMate is freely available for free non-commercial use under Apache License 2.0.

About

Virtual clothing try on tool powered by deep learning.

License:Apache License 2.0


Languages

Language:Python 56.4%Language:HTML 40.5%Language:Dockerfile 1.5%Language:Shell 1.0%Language:CSS 0.6%