Downloads · 30 days
0
adityavithaldas/Fashion_Category_Classifier
Fashion_Category_Classifier is a machine learning model from adityavithaldas. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc-by-4.0.
This model uses the Deep Fashion dataset in order to create a category classifier among the 50 or so provided categories. https://mmlab.ie.cuhk.edu.hk/projects/DeepFashion.html
Downloads · 30 days
0
Access
Public
Updated Mar 16, 2022
Repo size
—
Likes
5
Public
Click a slice to open those files.
Other1.2 KB · 64%
From the Hugging Face model README
This model uses the Deep Fashion dataset in order to create a category classifier among the 50 or so provided categories. https://mmlab.ie.cuhk.edu.hk/projects/DeepFashion.html
This model leverages the ViT (Vision transformer), loaded with the custom dataset and the 50 odd categoes to which they are assigned. The objective here, is to expand the same and get to a. An accuracy level of 90+ in the top 5 categorizes b. An accuracy of 70+ overall.
In addition, we would also look forward to creating attribute extractors, to extract key attributes (primary color, checked, sleeve, collar etc) as we proceed