A standalone HTTP web server that can recommend similar fashion outfits.
Uses multiple neural networks (with a ResNet50 backbone) behind the scenes to classify inputs by {category, texture, fabric, parts, shape}. The resulting embeddings are then used to query a pre-built nearest neighbors index for similar outputs.
Use pip to install the requirements.
pip install -r requirements.txt
To run the web server, simply execute flask with the main recommender app:
flask run
The main predictor can also be used independently of Flask, by calling get_recs
:
from predict import Predict
fashion = Predict()
recs = fashion.get_recs(img_path)
- Outfits - Example images that can be used to test the recommendation system
- app.py - Spins up a Flask App to serve recommendations
- predict.py - Recommendation System
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.