vkmanojk / VirtualTrialRoom

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

An e-commerce service with additional try-on facility where the user can witness how he/she would look if he/she wears the selected apparel.

Requisites

Setup

First clone the repository:

git clone https://github.com/vkmanojk/VirtualTrialRoom.git
cd VirtualTrialRoom

Next, run setup.sh, which will create a conda environment and install the required packages via

conda create --name fashionfiesta python=2.7 pip
source activate fashionfiesta
pip install -r requirements.txt

Additionally, setup.sh will clone and setup auxiliary repositories that do the Human parsing, Pose estimation, and Clothing style transfer steps. This process will download models for each of the tasks.

The e-commerce website designed for this project is available here Fashion Fiesta. This website was built using Flask and MySQL

Test Example

To test whether the setup was successful, run the following command that should reproduce the clothing transfer shown at the top of the README:

./run.sh inputs/example_person.jpg inputs/example_clothing.jpg

The corresponding output is output/output.png and should be similar to output/example_output.png (the right-most image above).

Notebooks

This repo includes some jupyter notebooks (located in notebooks/) that may be useful for understanding the SmartFit pipeline and its inputs/outputs. Exact usage instructions are at the top of each notebook.

  • visualize_inputs.ipynb
    • Explains and shows the main inputs to the clothing transfer model: body mask, face and pants, keypoint pose map, and clothing item.
  • visualize_output.ipynb
    • Displays the output image.
  • skintone_check.ipynb
    • Describes how the skintone check works. This is a check to make sure that the clothing transfer model did not change the skintone of the inputted person.

References

This project builds from the work listed below:

About


Languages

Language:Jupyter Notebook 96.9%Language:HTML 2.3%Language:Shell 0.5%Language:Python 0.4%