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