| { | |
| "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" | |
| } | |