NurzatS commited on
Commit
3f232a7
·
verified ·
1 Parent(s): 5bcd410

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +54 -0
README.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - ru
4
+ license: mit
5
+ pipeline_tag: image-classification
6
+ tags:
7
+ - vision
8
+ - image-classification
9
+ - pytorch
10
+ - vit
11
+ - e-commerce
12
+ metrics:
13
+ - accuracy
14
+ model-index:
15
+ - name: Product_Image_Classification
16
+ results:
17
+ - task:
18
+ type: image-classification
19
+ metrics:
20
+ - type: accuracy
21
+ value: 0.9448
22
+ ---
23
+
24
+ # Product Image Classification
25
+
26
+ Классификатор изображений товаров (смартфоны, ноутбуки, телевизоры, часы, наушники) на базе **Vision Transformer (ViT)**.
27
+
28
+ Модель обучена на датасете изображений, собранных с узбекских маркетплейсов (*asaxiy.uz*, *texnomart.uz*, *olcha.uz*).
29
+
30
+ ## Описание и характеристики
31
+
32
+ * **Базовая архитектура:** `google/vit-base-patch16-224-in21k`
33
+ * **Категории (5 классов):** `headphones`, `laptop`, `smartphone`, `tv`, `watch`
34
+ * **Размер входного изображения:** 224x224 RGB
35
+
36
+ ## Качество модели (Evaluation)
37
+
38
+ Оценка производилась на тестовой выборке (`dataset_prepared/test`, 163 изображения):
39
+
40
+ | Категория | Precision | Recall | F1-Score | Support |
41
+ | :--- | :---: | :---: | :---: | :---: |
42
+ | **headphones** | 0.9091 | 0.9091 | 0.9091 | 33 |
43
+ | **laptop** | 0.9655 | 0.9032 | 0.9333 | 31 |
44
+ | **smartphone** | 0.8684 | 1.0000 | 0.9296 | 33 |
45
+ | **tv** | 1.0000 | 0.9697 | 0.9846 | 33 |
46
+ | **watch** | 1.0000 | 0.9394 | 0.9688 | 33 |
47
+ | **Accuracy** | | | **0.9448** | **163** |
48
+
49
+ ## Быстрый старт (Quickstart)
50
+
51
+ Для использования модели вам понадобятся библиотеки `transformers`, `torch` и `pillow`:
52
+
53
+ ```bash
54
+ pip install transformers torch pillow