Hugging Face's logo

meet67
/
Egde-color-classification

Image Classification
Transformers
PyTorch
TensorBoard
huggingpics
Eval Results
Inference Endpoints
Model card Files Metrics Community
meet67 commited on
Commit
5ab4865
•
1 Parent(s): 633cf01

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -19
config.json CHANGED
@@ -34,13 +34,7 @@
34
  224,
35
  448
36
  ],
37
- "id2label": {
38
- "0": "LABEL_0",
39
- "1": "LABEL_1",
40
- "2": "LABEL_2",
41
- "3": "LABEL_3",
42
- "4": "LABEL_4"
43
- },
44
  "image_size": 224,
45
  "initializer_range": 0.02,
46
  "input_size": [
@@ -50,19 +44,14 @@
50
  ],
51
  "interpolation": "bicubic",
52
  "key_dim": 32,
53
- "label2id": {
54
- "LABEL_0": 0,
55
- "LABEL_1": 1,
56
- "LABEL_2": 2,
57
- "LABEL_3": 3,
58
- "LABEL_4": 4
59
- },
60
  "labels": [
61
- "LABEL_0",
62
- "LABEL_1",
63
- "LABEL_2",
64
- "LABEL_3",
65
- "LABEL_4",
66
  "LABEL_5",
67
  "LABEL_6",
68
  "LABEL_7",
 
34
  224,
35
  448
36
  ],
37
+ "id2label": {'0': 'Black', '1': 'Blue', '2': 'green', '3': 'red', '4': 'white'},
 
 
 
 
 
 
38
  "image_size": 224,
39
  "initializer_range": 0.02,
40
  "input_size": [
 
44
  ],
45
  "interpolation": "bicubic",
46
  "key_dim": 32,
47
+ "label2id":
48
+ {'Black': '0', 'Blue': '1', 'green': '2', 'red': '3', 'white': '4'},
 
 
 
 
 
49
  "labels": [
50
+ "Black",
51
+ "Blue",
52
+ "green",
53
+ "red",
54
+ "white",
55
  "LABEL_5",
56
  "LABEL_6",
57
  "LABEL_7",