mstatt commited on
Commit
465e6d4
·
verified ·
1 Parent(s): 683913b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -3
README.md CHANGED
@@ -60,8 +60,14 @@ The base model used is the Vision Transformer `vit-base-patch16-224-in21k`, whic
60
  The model was trained on a curated dataset of UAE company logos as well as others of international companies. The dataset consists of thousands of images across various brands to ensure robustness and accuracy.
61
 
62
  ### Performance
63
- The model achieved high accuracy on a held-out validation set, indicating strong performance in classifying UAE company logos. Detailed performance metrics (accuracy, precision, recall, F1-score) can be provided upon request.
64
-
 
 
 
 
 
 
65
  ## How to Use
66
  To use the model for inference, you can load it using the `transformers` library from Hugging Face:
67
 
@@ -154,7 +160,7 @@ print(model.config.id2label[predicted_label])
154
 
155
 
156
  ### Limitations and Biases
157
- - The model is specifically trained on UAE company logos and may not perform well on logos from companies outside the UAE.
158
  - The model's performance is contingent upon the quality and diversity of the training dataset.
159
  - Potential biases in the training data can lead to biases in model predictions.
160
 
 
60
  The model was trained on a curated dataset of UAE company logos as well as others of international companies. The dataset consists of thousands of images across various brands to ensure robustness and accuracy.
61
 
62
  ### Performance
63
+ ***** eval metrics *****
64
+ - epoch = 20.0
65
+ - eval_accuracy = 0.9761
66
+ - eval_loss = 0.1193
67
+ - eval_runtime = 0:00:20.51
68
+ - eval_samples_per_second = 268.951
69
+ - eval_steps_per_second = 8.432
70
+
71
  ## How to Use
72
  To use the model for inference, you can load it using the `transformers` library from Hugging Face:
73
 
 
160
 
161
 
162
  ### Limitations and Biases
163
+ - The model is specifically trained on UAE company logos and a few from Outside the UAE, it may not perform well on logos from numerous other large or small companies.
164
  - The model's performance is contingent upon the quality and diversity of the training dataset.
165
  - Potential biases in the training data can lead to biases in model predictions.
166