Skip to content

Fashion Tagging

Overview

Visual features are enough for the models to identify trend and rank them. However they are not interpretable. We developed a fashion tagging model which can tag images with fashion attributes, there are about 1000 tags in the training dataset we used

Tagging can help the user interpret the results of the ranking algorithm much better Dashboards can be made where highly ranked fashion images and their corresponding tags can be used to give insightful plots

Moreover these tags can also be used for textual search, where the user can search whether a particular fashion is in trend by using words.

Dataset

We used DeepFashion Attribute prediction dataset. This dataset had about 280,000 images belonging to 5000 classes, each class had its own unique fashion style

Training

Model

We used ResNet model as the backbone, the last layer was a 1000 units linear layer with sigmoid activation function.

Loss

We used the cross entropy loss

Scope For Improvement

We can make make an end-to-end model with bbox regressor and attribute classifier to improve the accuracy of the model.