/OutfitGen

Submission to Myntra Hackerramp: Campus Edition 2021 (Theme: Enhance)

Primary LanguagePython

Outfit Generator

Submission to Myntra Hackerramp: Campus Edition 2021 (Theme: Enhance)

App GIF

Usage

  1. We use the Polyvore dataset. In fact we use a cleaner version of the dataset, that is available in this repo.

  2. Extract the polyvore-images dataset into an image folder in the data directory.

  3. To train or evaluate the model, go into the model directory and run train.py or evaluate.py respectively. (to evaluate you must place the trained model in the model directory with the name Final.pth)

  4. To run the web app, go into the App directory and use

    streamlit run OutfitGenerator.py
    

    to run the app on your local browser

Pretrained Model

Download the pretrained model and place it in the model directory to use.

Dependencies

  • pytorch
  • torchvision
  • scikit-learn
  • matplotlib
  • streamlit