rrighart commited on
Commit
32eb0f4
1 Parent(s): ba089ac

Updates text

Browse files
Files changed (1) hide show
  1. README.md +34 -12
README.md CHANGED
@@ -1,12 +1,34 @@
1
- ---
2
- title: Color Tags
3
- emoji: 🌍
4
- colorFrom: yellow
5
- colorTo: indigo
6
- sdk: gradio
7
- sdk_version: 3.2
8
- app_file: app.py
9
- pinned: false
10
- ---
11
-
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ # Color tags
3
+
4
+ ## An App classifying images on different colors
5
+
6
+ App: https://huggingface.co/spaces/rrighart/color-tags
7
+
8
+ Color tags are very useful for finding pictures on the web, such as in e-commerce applications.
9
+ This App provides color tags for uploaded image files.
10
+ The underlying model was trained using a basic Convolutional Neural Network and an EfficientNet, with the latter giving slightly better results [[1]](https://www.kaggle.com/code/rrighart/the-prediction-of-color-tags/data) .
11
+
12
+ For training and validation, the Vehicles Color Recognition (VCoR) dataset was used, which has 15 colors classes [[2]](https://www.kaggle.com/datasets/landrykezebou/vcor-vehicle-color-recognition-dataset). A separate unseen testset has overall accuracy of 84%.
13
+ 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]](https://www.kaggle.com/datasets/paramaggarwal/fashion-product-images-small) which shows promising results.
14
+
15
+ ## Contact
16
+
17
+ Ruthger Righart (PhD)
18
+
19
+ Data scientist in machine learning and computer vision
20
+
21
+ Email: rrighart@googlemail.com
22
+
23
+ Web: https://www.rrighart.com
24
+
25
+ ## References
26
+
27
+ 1. The prediction of color tags, Kaggle Notebook R. Righart.
28
+ 2. Vehicle color recognition dataset (VCoR).
29
+ 3. Fashion Product Images.
30
+
31
+
32
+
33
+
34
+