Cloth Recommendation System

Convolutional Neural Network for recommending cloths

Creating a custom model to train still remaining due to lack of resources for now

Output from a VGG19 (pre-trained)

Input image

Returned similar images

Input image

Returned similar images

Input image

Returned similar images

Install the dependencies.

$ pip install -r requirements.txt

Augment images

$ python augment_images.py

To train the model.

$ python train.py

For tensorboard vizualization while training.

$ tensorboard --logdir=tensorboard_logs --host=localhost --port=8088

Note:

Different model require different size if images. This can be edited in the tf_records.py in resize opereation.