https://github.com/khanhnamle1994/fashion-recommendation

A clothing retrieval and visual recommendation model for fashion images.
https://github.com/khanhnamle1994/fashion-recommendation

computer-vision deepfashion fashion-recommendation floydhub numpy pandas python3 resnet tensorflow

Last synced: 7 days ago
JSON representation

A clothing retrieval and visual recommendation model for fashion images.

• Host: GitHub
• URL: https://github.com/khanhnamle1994/fashion-recommendation
• Owner: khanhnamle1994
• Created: 2019-01-11T12:28:08.000Z (almost 6 years ago)
• Default Branch: master
• Last Pushed: 2019-02-08T20:47:21.000Z (almost 6 years ago)
• Last Synced: 2024-12-13T17:41:32.116Z (18 days ago)
• Topics: computer-vision, deepfashion, fashion-recommendation, floydhub, numpy, pandas, python3, resnet, tensorflow
• Language: Python
• Size: 23.4 MB
• Stars: 319
• Watchers: 19
• Forks: 100
• Open Issues: 5
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Fashion Recommendation

In this project, I created an end-to-end solution for large-scale image classification and visual recommendation on fashion images. More specifically, my model can learn the important regions in an image and generate diverse recommendations based on such semantic similarity.  

First, I built a clothing image classification model using a ResNet-based model. The feature layer of this model can capture fine-grained semantic clothing features like fabrics, styles and patterns of the clothes. Then, using such features, the model can recommend similar clothes to the input images using nearest neighbor search.

An accompanied blog post via FloydHub can be found at this link: [Recommending Similar Fashion Images with Deep Learning](https://blog.floydhub.com/similar-fashion-images/)

## Dataset

[DeepFashion](http://mmlab.ie.cuhk.edu.hk/projects/DeepFashion.html) is a large-scale clothes database that is quite popular in the research community. It contains over 800,000 diverse fashion images ranging from well-posed shop images to unconstrained consumer photos. It is annotated with rich information of clothing items. It also contains over 300,000 cross-pose/cross-domain image pairs.

More specifically, I used the images in the [DeepFashion Attribute Prediction](http://mmlab.ie.cuhk.edu.hk/projects/DeepFashion/AttributePrediction.html) subset. It contains 289,222 number of clothes images, 50 number of clothing categories, and 1,000 number of clothing attributes. Each image is annotated by bounding box and clothing type.

![AttributePrediction](deep-fashion-attribute-prediction.jpg)

## Code

1. [preprocessing.py](https://github.com/khanhnamle1994/fashion-recommendation/blob/master/code/preprocessing.py): This code is used to pre-process the dataset.

2. [hyper_parameters.py](https://github.com/khanhnamle1994/fashion-recommendation/blob/master/code/hyper_parameters.py): This code is used to define all hyper-parameters regarding training.

3. [fashion_input.py](https://github.com/khanhnamle1994/fashion-recommendation/blob/master/code/fashion_input.py): This code is used to pre-process the image further during training.

4. [simple_resnet.py](https://github.com/khanhnamle1994/fashion-recommendation/blob/master/code/simple_resnet.py): This code is used to define ResNet architecture.

5. [train_n_test.py](https://github.com/khanhnamle1994/fashion-recommendation/blob/master/code/train_n_test.py): This code is used as the main training file.

## Sample Results

1 - Romper Category

![Romper](Results/Romper-Examples.png)

2 - Jacket Category

![Jacket](Results/Jacket-Examples.png)

3 - Blouse Shirt Category

![BlouseShirt](Results/BlouseShirt-Examples.png)

## Requirements

- [TensorFlow](https://www.tensorflow.org/) latest version

- [pandas](https://pandas.pydata.org/) latest version

- [numpy](http://www.numpy.org/) latest version

## Run on FloydHub

[![Run on FloydHub](https://static.floydhub.com/button/button-small.svg)](https://floydhub.com/run)