config.json
1.6 KB · 112 lines · json Raw
1 {
2 "_name_or_path": "jonathandinu/face-parsing",
3 "architectures": [
4 "SegformerForSemanticSegmentation"
5 ],
6 "attention_probs_dropout_prob": 0.0,
7 "classifier_dropout_prob": 0.1,
8 "decoder_hidden_size": 768,
9 "depths": [
10 3,
11 6,
12 40,
13 3
14 ],
15 "downsampling_rates": [
16 1,
17 4,
18 8,
19 16
20 ],
21 "drop_path_rate": 0.1,
22 "hidden_act": "gelu",
23 "hidden_dropout_prob": 0.0,
24 "hidden_sizes": [
25 64,
26 128,
27 320,
28 512
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": [
76 4,
77 4,
78 4,
79 4
80 ],
81 "model_type": "segformer",
82 "num_attention_heads": [
83 1,
84 2,
85 5,
86 8
87 ],
88 "num_channels": 3,
89 "num_encoder_blocks": 4,
90 "patch_sizes": [
91 7,
92 3,
93 3,
94 3
95 ],
96 "reshape_last_stage": true,
97 "semantic_loss_ignore_index": 255,
98 "sr_ratios": [
99 8,
100 4,
101 2,
102 1
103 ],
104 "strides": [
105 4,
106 2,
107 2,
108 2
109 ],
110 "transformers_version": "4.37.0.dev0"
111 }
112