Upload 5 files
Browse files- config.json +21 -0
- onnx/model.onnx +3 -0
- onnx/model_quantized.onnx +3 -0
- preprocessor_config.json +18 -0
- quantize_config.json +19 -0
config.json
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
|
3 |
+
"num_labels": 5,
|
4 |
+
"id2label": {
|
5 |
+
"0": "Brick",
|
6 |
+
"1": "Metal",
|
7 |
+
"2": "Paper",
|
8 |
+
"3": "Plastic",
|
9 |
+
"4": "Wood"
|
10 |
+
},
|
11 |
+
"label2id": {
|
12 |
+
"Brick": 0,
|
13 |
+
"Metal": 1,
|
14 |
+
"Paper": 2,
|
15 |
+
"Plastic": 3,
|
16 |
+
"Wood": 4
|
17 |
+
},
|
18 |
+
"model_type": "resnet",
|
19 |
+
"num_channels": 3,
|
20 |
+
"transformers_version": "4.42.4"
|
21 |
+
}
|
onnx/model.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5fc26805177f8dd47fe87a46eec51e12959a7ceccc1ce3cd7c90e91b4c38d2ea
|
3 |
+
size 98684032
|
onnx/model_quantized.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:71e74b0b4e66dea7b6e204540e43b2a22e75660888cf17cdee7e534ba34bd94d
|
3 |
+
size 24887086
|
preprocessor_config.json
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"do_normalize": true,
|
3 |
+
"do_resize": true,
|
4 |
+
"image_mean": [
|
5 |
+
0.485,
|
6 |
+
0.456,
|
7 |
+
0.406
|
8 |
+
],
|
9 |
+
"image_std": [
|
10 |
+
0.229,
|
11 |
+
0.224,
|
12 |
+
0.225
|
13 |
+
],
|
14 |
+
"size": {
|
15 |
+
"height": 224,
|
16 |
+
"width": 224
|
17 |
+
}
|
18 |
+
}
|
quantize_config.json
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"per_channel": true,
|
3 |
+
"reduce_range": true,
|
4 |
+
"per_model_config": {
|
5 |
+
"model": {
|
6 |
+
"op_types": [
|
7 |
+
"MaxPool",
|
8 |
+
"Identity",
|
9 |
+
"Flatten",
|
10 |
+
"Relu",
|
11 |
+
"GlobalAveragePool",
|
12 |
+
"Add",
|
13 |
+
"Conv",
|
14 |
+
"Gemm"
|
15 |
+
],
|
16 |
+
"weight_type": "QUInt8"
|
17 |
+
}
|
18 |
+
}
|
19 |
+
}
|