pogzyb commited on
Commit
ebb3cb9
1 Parent(s): 95aeae9

Upload folder using huggingface_hub

Browse files
.ipynb_checkpoints/config-checkpoint.json ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "attention_probs_dropout_prob": 0.0,
3
+ "classifier_dropout_prob": 0.1,
4
+ "decoder_hidden_size": 256,
5
+ "depths": [
6
+ 2,
7
+ 2,
8
+ 2,
9
+ 2
10
+ ],
11
+ "drop_path_rate": 0.1,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.0,
14
+ "hidden_sizes": [
15
+ 32,
16
+ 64,
17
+ 160,
18
+ 256
19
+ ],
20
+ "id2label": {
21
+ "0": "skin",
22
+ "1": "l_brow",
23
+ "2": "r_brow",
24
+ "3": "l_eye",
25
+ "4": "r_eye",
26
+ "5": "eye_g",
27
+ "6": "l_ear",
28
+ "7": "r_ear",
29
+ "8": "ear_r",
30
+ "9": "nose",
31
+ "10": "mouth",
32
+ "11": "u_lip",
33
+ "12": "l_lip",
34
+ "13": "neck",
35
+ "14": "neck_l",
36
+ "15": "cloth",
37
+ "16": "hair",
38
+ "17": "hat"
39
+ },
40
+ "initializer_range": 0.02,
41
+ "label2id": {
42
+ "cloth": 15,
43
+ "ear_r": 8,
44
+ "eye_g": 5,
45
+ "hair": 16,
46
+ "hat": 17,
47
+ "l_brow": 1,
48
+ "l_ear": 6,
49
+ "l_eye": 3,
50
+ "l_lip": 12,
51
+ "mouth": 10,
52
+ "neck": 13,
53
+ "neck_l": 14,
54
+ "nose": 9,
55
+ "r_brow": 2,
56
+ "r_ear": 7,
57
+ "r_eye": 4,
58
+ "skin": 0,
59
+ "u_lip": 11
60
+ },
61
+ "layer_norm_eps": 1e-06,
62
+ "mlp_ratios": [
63
+ 4,
64
+ 4,
65
+ 4,
66
+ 4
67
+ ],
68
+ "model_type": "segformer",
69
+ "num_attention_heads": [
70
+ 1,
71
+ 2,
72
+ 5,
73
+ 8
74
+ ],
75
+ "num_channels": 3,
76
+ "num_classes": 18,
77
+ "num_encoder_blocks": 4,
78
+ "patch_sizes": [
79
+ 7,
80
+ 3,
81
+ 3,
82
+ 3
83
+ ],
84
+ "reshape_last_stage": true,
85
+ "semantic_loss_ignore_index": 255,
86
+ "sr_ratios": [
87
+ 8,
88
+ 4,
89
+ 2,
90
+ 1
91
+ ],
92
+ "strides": [
93
+ 4,
94
+ 2,
95
+ 2,
96
+ 2
97
+ ],
98
+ "transformers_version": "4.36.2"
99
+ }
onnx/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24c634d65a0e6c4cf654108fc294614ef7482aeea0d38e184c7a6e286b942322
3
+ size 15127595
onnx/model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a303b665aec041814d0af50360573fbb58fa6520c7e8d5df6fc7c98496a710d
3
+ size 4320146
quantize_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "per_channel": true,
3
+ "reduce_range": true,
4
+ "per_model_config": {
5
+ "model": {
6
+ "op_types": [
7
+ "Add",
8
+ "Relu",
9
+ "MatMul",
10
+ "Reshape",
11
+ "Gather",
12
+ "Identity",
13
+ "Transpose",
14
+ "Slice",
15
+ "Cast",
16
+ "Sqrt",
17
+ "Unsqueeze",
18
+ "Resize",
19
+ "Erf",
20
+ "Shape",
21
+ "ReduceMean",
22
+ "Mul",
23
+ "Sub",
24
+ "Concat",
25
+ "Softmax",
26
+ "Conv",
27
+ "Constant",
28
+ "Div",
29
+ "Pow"
30
+ ],
31
+ "weight_type": "QUInt8"
32
+ }
33
+ }
34
+ }