face-segmenter / config.json
pogzyb's picture
Upload SegformerForSemanticSegmentation
2163be8 verified
{
"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"
}