felixwf commited on
Commit
1309526
·
verified ·
1 Parent(s): 4c5a3e4

Upload ViTForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +17 -20
  2. model.safetensors +2 -2
config.json CHANGED
@@ -9,33 +9,30 @@
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "angry",
13
  "1": "disgust",
14
- "2": "fear",
15
- "3": "frowning",
16
- "4": "happy",
17
- "5": "kiss",
18
- "6": "neutral",
19
- "7": "openmouth",
20
- "8": "raisedbrows",
21
- "9": "sad",
22
- "10": "surprise"
23
  },
24
  "image_size": 224,
25
  "initializer_range": 0.02,
26
  "intermediate_size": 3072,
27
  "label2id": {
28
- "angry": 0,
29
  "disgust": 1,
30
- "fear": 2,
31
- "frowning": 3,
32
- "happy": 4,
33
- "kiss": 5,
34
- "neutral": 6,
35
- "openmouth": 7,
36
- "raisedbrows": 8,
37
- "sad": 9,
38
- "surprise": 10
 
39
  },
40
  "layer_norm_eps": 1e-12,
41
  "model_type": "vit",
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "sad",
13
  "1": "disgust",
14
+ "2": "angry",
15
+ "3": "neutral",
16
+ "4": "fear",
17
+ "5": "surprise",
18
+ "6": "happy"
 
 
 
 
19
  },
20
  "image_size": 224,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
+ "angry": 2,
25
  "disgust": 1,
26
+ "fear": 4,
27
+ "frowning": 7,
28
+ "happy": 6,
29
+ "kiss": 8,
30
+ "neutral": 3,
31
+ "openmouth": 9,
32
+ "raisedbrows": 10,
33
+ "sad": 0,
34
+ "smile": 11,
35
+ "surprise": 5
36
  },
37
  "layer_norm_eps": 1e-12,
38
  "model_type": "vit",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:42a8c2db10c7002f88b07990de9b919bef7e969de86336933657839358b54177
3
- size 343251660
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c3ea033cea51addaa9fdae55bbdf1142a7ae4f37ca2842fcfea7ca36dcaa05bd
3
+ size 343239356