File size: 999 Bytes
7c916ae 2163be8 7c916ae 2163be8 7c916ae 2163be8 7c916ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
{
"architectures": [
"SegformerForSemanticSegmentation"
],
"auto_map": {
"AutoConfig": "model.FaceSegmenterConfig",
"AutoModelForImageSegmentation": "model.SegformerForSemanticSegmentation"
},
"id2label": {
"0": "skin",
"1": "l_brow",
"2": "r_brow",
"3": "l_eye",
"4": "r_eye",
"5": "eye_g",
"6": "l_ear",
"7": "r_ear",
"8": "ear_r",
"9": "nose",
"10": "mouth",
"11": "u_lip",
"12": "l_lip",
"13": "neck",
"14": "neck_l",
"15": "cloth",
"16": "hair",
"17": "hat"
},
"label2id": {
"cloth": 15,
"ear_r": 8,
"eye_g": 5,
"hair": 16,
"hat": 17,
"l_brow": 1,
"l_ear": 6,
"l_eye": 3,
"l_lip": 12,
"mouth": 10,
"neck": 13,
"neck_l": 14,
"nose": 9,
"r_brow": 2,
"r_ear": 7,
"r_eye": 4,
"skin": 0,
"u_lip": 11
},
"model_type": "image-segmentation",
"num_classes": 18,
"torch_dtype": "float32",
"transformers_version": "4.37.0"
}
|