Fix labels

config.json

@@ -29,47 +29,47 @@
   ],
   "id2label": {
     "0": "background",
-    "1": "skin.png",
-    "2": "nose.png",
-    "3": "r_eye.png",
-    "4": "l_eye.png",
-    "5": "r_brow.png",
-    "6": "l_brow.png",
-    "7": "r_ear.png",
-    "8": "l_ear.png",
-    "9": "mouth.png",
-    "10": "u_lip.png",
-    "11": "l_lip.png",
-    "12": "neck.png",
-    "13": "hair.png",
-    "14": "neck_l.png",
-    "15": "cloth.png",
-    "16": "eye_g.png",
-    "17": "hat.png",
-    "18": "ear_r.png"
+    "1": "skin",
+    "2": "nose",
+    "3": "eye_g",
+    "4": "l_eye",
+    "5": "r_eye",
+    "6": "l_brow",
+    "7": "r_brow",
+    "8": "l_ear",
+    "9": "r_ear",
+    "10": "mouth",
+    "11": "u_lip",
+    "12": "l_lip",
+    "13": "hair",
+    "14": "hat",
+    "15": "ear_r",
+    "16": "neck_l",
+    "17": "neck",
+    "18": "cloth"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
     "background": 0,
-    "cloth.png": 15,
-    "ear_r.png": 18,
-    "eye_g.png": 16,
-    "hair.png": 13,
-    "hat.png": 17,
-    "l_brow.png": 6,
-    "l_ear.png": 8,
-    "l_eye.png": 4,
-    "l_lip.png": 11,
-    "mouth.png": 9,
-    "neck.png": 12,
-    "neck_l.png": 14,
-    "nose.png": 2,
-    "r_brow.png": 5,
-    "r_ear.png": 7,
-    "r_eye.png": 3,
-    "skin.png": 1,
-    "u_lip.png": 10
+    "skin": 1,
+    "nose": 2,
+    "eye_g": 3,
+    "l_eye": 4,
+    "r_eye": 5,
+    "l_brow": 6,
+    "r_brow": 7,
+    "l_ear": 8,
+    "r_ear": 9,
+    "mouth": 10,
+    "u_lip": 11,
+    "l_lip": 12,
+    "hair": 13,
+    "hat": 14,
+    "ear_r": 15,
+    "neck_l": 16,
+    "neck": 17,
+    "cloth": 18
   },
   "layer_norm_eps": 1e-06,
   "mlp_ratios": [