This is an project for hackthon version beta from error_404
- Conda
- Octave 4.2+ (with image package)
- Also compatible with MATLAB
First clone the repository:
Next, run source setup.sh, which will create a conda environment and install the required packages via
conda create --name smartfit python=2.7 pip
source activate smartfit
pip install -r requirements.txt
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_smartfit.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).
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.
This project builds from the work listed below: