jonathandinu commited on
Commit
1f7e152
1 Parent(s): 373bfce

add onnx weights for transformers.js

Browse files
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/Users/jonathandinu/Desktop/PROJECTS/arch_fake_faces/ethereum/gcp/assets/segformer_b5",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
@@ -107,6 +107,5 @@
107
  2,
108
  2
109
  ],
110
- "torch_dtype": "float32",
111
- "transformers_version": "4.20.1"
112
  }
 
1
  {
2
+ "_name_or_path": "jonathandinu/face-parsing",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
 
107
  2,
108
  2
109
  ],
110
+ "transformers_version": "4.37.0.dev0"
 
111
  }
onnx/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d4e67af60ff78184745ebf74cc15163c0adc27d45cdeba31e3a03d1096fb8c3
3
+ size 340316611
onnx/model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5bab9bfb3cb979f3098ac3b934b1641dbf87f835e0b03c2ca6d88dcf18c83d27
3
+ size 89439678
preprocessor_config.json CHANGED
@@ -1,18 +1,23 @@
1
  {
2
  "do_normalize": true,
 
 
3
  "do_resize": true,
4
- "feature_extractor_type": "SegformerFeatureExtractor",
5
  "image_mean": [
6
  0.485,
7
  0.456,
8
  0.406
9
  ],
 
10
  "image_std": [
11
  0.229,
12
  0.224,
13
  0.225
14
  ],
15
- "reduce_labels": false,
16
  "resample": 2,
17
- "size": 512
 
 
 
 
18
  }
 
1
  {
2
  "do_normalize": true,
3
+ "do_reduce_labels": false,
4
+ "do_rescale": true,
5
  "do_resize": true,
 
6
  "image_mean": [
7
  0.485,
8
  0.456,
9
  0.406
10
  ],
11
+ "image_processor_type": "SegformerFeatureExtractor",
12
  "image_std": [
13
  0.229,
14
  0.224,
15
  0.225
16
  ],
 
17
  "resample": 2,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 512,
21
+ "width": 512
22
+ }
23
  }
quantize_config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "per_channel": true,
3
+ "reduce_range": true,
4
+ "per_model_config": {
5
+ "model": {
6
+ "op_types": [
7
+ "Unsqueeze",
8
+ "Shape",
9
+ "Transpose",
10
+ "Sqrt",
11
+ "Gather",
12
+ "Slice",
13
+ "Erf",
14
+ "Div",
15
+ "Reshape",
16
+ "Add",
17
+ "Cast",
18
+ "Sub",
19
+ "Concat",
20
+ "ReduceMean",
21
+ "Mul",
22
+ "Conv",
23
+ "Constant",
24
+ "Resize",
25
+ "Softmax",
26
+ "Pow",
27
+ "Relu",
28
+ "MatMul"
29
+ ],
30
+ "weight_type": "QUInt8"
31
+ }
32
+ }
33
+ }