Xenova HF staff commited on
Commit
7f98ba7
1 Parent(s): 8ee285c

Fix labels

Browse files
Files changed (1) hide show
  1. config.json +36 -36
config.json CHANGED
@@ -29,47 +29,47 @@
29
  ],
30
  "id2label": {
31
  "0": "background",
32
- "1": "skin.png",
33
- "2": "nose.png",
34
- "3": "r_eye.png",
35
- "4": "l_eye.png",
36
- "5": "r_brow.png",
37
- "6": "l_brow.png",
38
- "7": "r_ear.png",
39
- "8": "l_ear.png",
40
- "9": "mouth.png",
41
- "10": "u_lip.png",
42
- "11": "l_lip.png",
43
- "12": "neck.png",
44
- "13": "hair.png",
45
- "14": "neck_l.png",
46
- "15": "cloth.png",
47
- "16": "eye_g.png",
48
- "17": "hat.png",
49
- "18": "ear_r.png"
50
  },
51
  "image_size": 224,
52
  "initializer_range": 0.02,
53
  "label2id": {
54
  "background": 0,
55
- "cloth.png": 15,
56
- "ear_r.png": 18,
57
- "eye_g.png": 16,
58
- "hair.png": 13,
59
- "hat.png": 17,
60
- "l_brow.png": 6,
61
- "l_ear.png": 8,
62
- "l_eye.png": 4,
63
- "l_lip.png": 11,
64
- "mouth.png": 9,
65
- "neck.png": 12,
66
- "neck_l.png": 14,
67
- "nose.png": 2,
68
- "r_brow.png": 5,
69
- "r_ear.png": 7,
70
- "r_eye.png": 3,
71
- "skin.png": 1,
72
- "u_lip.png": 10
73
  },
74
  "layer_norm_eps": 1e-06,
75
  "mlp_ratios": [
 
29
  ],
30
  "id2label": {
31
  "0": "background",
32
+ "1": "skin",
33
+ "2": "nose",
34
+ "3": "eye_g",
35
+ "4": "l_eye",
36
+ "5": "r_eye",
37
+ "6": "l_brow",
38
+ "7": "r_brow",
39
+ "8": "l_ear",
40
+ "9": "r_ear",
41
+ "10": "mouth",
42
+ "11": "u_lip",
43
+ "12": "l_lip",
44
+ "13": "hair",
45
+ "14": "hat",
46
+ "15": "ear_r",
47
+ "16": "neck_l",
48
+ "17": "neck",
49
+ "18": "cloth"
50
  },
51
  "image_size": 224,
52
  "initializer_range": 0.02,
53
  "label2id": {
54
  "background": 0,
55
+ "skin": 1,
56
+ "nose": 2,
57
+ "eye_g": 3,
58
+ "l_eye": 4,
59
+ "r_eye": 5,
60
+ "l_brow": 6,
61
+ "r_brow": 7,
62
+ "l_ear": 8,
63
+ "r_ear": 9,
64
+ "mouth": 10,
65
+ "u_lip": 11,
66
+ "l_lip": 12,
67
+ "hair": 13,
68
+ "hat": 14,
69
+ "ear_r": 15,
70
+ "neck_l": 16,
71
+ "neck": 17,
72
+ "cloth": 18
73
  },
74
  "layer_norm_eps": 1e-06,
75
  "mlp_ratios": [