Spaces:
Runtime error
Color tags
An App classifying images on different colors
App: https://huggingface.co/spaces/rrighart/color-tags
Color tags are very useful for finding pictures on the web, such as in e-commerce applications. This App provides color tags for uploaded image files. The underlying model was trained using a basic Convolutional Neural Network and an EfficientNet, with the latter giving slightly better results [1] .
For training and validation, the Vehicles Color Recognition (VCoR) dataset was used, which has 15 colors classes [2]. A separate unseen testset has overall accuracy of 84%. One goal of this project is to see if the model can generalizes to other kind of products and objects. For this purpose tested some images from the Fashion Product Images data [3] which shows promising results.
Contact
Ruthger Righart (PhD)
Data scientist in machine learning and computer vision
Email: rrighart@googlemail.com
References
- The prediction of color tags, Kaggle Notebook R. Righart.
- Vehicle color recognition dataset (VCoR).
- Fashion Product Images.