Marc commited on
Commit
f8ae376
·
1 Parent(s): fcd9a6e

fixing cannot load model issue

Browse files
config.json CHANGED
@@ -1,27 +1,167 @@
1
  {
2
- "model": {
3
- "xml": "model.xml",
4
- "bin": "model.bin"
5
- },
6
- "nn_config": {
7
- "output_format": "detection",
8
- "NN_family": "YOLO",
9
- "input_size": "640x640",
10
- "NN_specific_metadata": {
11
- "classes": 2,
12
- "coordinates": 4,
13
- "anchors": [],
14
- "anchor_masks": {},
15
- "iou_threshold": 0.5,
16
- "confidence_threshold": 0.5
17
- }
18
- },
19
- "mappings": {
20
- "labels": [
21
- "person_body",
22
- "person_head"
23
- ]
24
- },
25
- "version": 1,
26
- "model_type": "yolov10"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
  }
 
1
  {
2
+ "id2label": {
3
+ "0": "person",
4
+ "1": "bicycle",
5
+ "2": "car",
6
+ "3": "motorcycle",
7
+ "4": "airplane",
8
+ "5": "bus",
9
+ "6": "train",
10
+ "7": "truck",
11
+ "8": "boat",
12
+ "9": "traffic light",
13
+ "10": "fire hydrant",
14
+ "11": "stop sign",
15
+ "12": "parking meter",
16
+ "13": "bench",
17
+ "14": "bird",
18
+ "15": "cat",
19
+ "16": "dog",
20
+ "17": "horse",
21
+ "18": "sheep",
22
+ "19": "cow",
23
+ "20": "elephant",
24
+ "21": "bear",
25
+ "22": "zebra",
26
+ "23": "giraffe",
27
+ "24": "backpack",
28
+ "25": "umbrella",
29
+ "26": "handbag",
30
+ "27": "tie",
31
+ "28": "suitcase",
32
+ "29": "frisbee",
33
+ "30": "skis",
34
+ "31": "snowboard",
35
+ "32": "sports ball",
36
+ "33": "kite",
37
+ "34": "baseball bat",
38
+ "35": "baseball glove",
39
+ "36": "skateboard",
40
+ "37": "surfboard",
41
+ "38": "tennis racket",
42
+ "39": "bottle",
43
+ "40": "wine glass",
44
+ "41": "cup",
45
+ "42": "fork",
46
+ "43": "knife",
47
+ "44": "spoon",
48
+ "45": "bowl",
49
+ "46": "banana",
50
+ "47": "apple",
51
+ "48": "sandwich",
52
+ "49": "orange",
53
+ "50": "broccoli",
54
+ "51": "carrot",
55
+ "52": "hot dog",
56
+ "53": "pizza",
57
+ "54": "donut",
58
+ "55": "cake",
59
+ "56": "chair",
60
+ "57": "couch",
61
+ "58": "potted plant",
62
+ "59": "bed",
63
+ "60": "dining table",
64
+ "61": "toilet",
65
+ "62": "tv",
66
+ "63": "laptop",
67
+ "64": "mouse",
68
+ "65": "remote",
69
+ "66": "keyboard",
70
+ "67": "cell phone",
71
+ "68": "microwave",
72
+ "69": "oven",
73
+ "70": "toaster",
74
+ "71": "sink",
75
+ "72": "refrigerator",
76
+ "73": "book",
77
+ "74": "clock",
78
+ "75": "vase",
79
+ "76": "scissors",
80
+ "77": "teddy bear",
81
+ "78": "hair drier",
82
+ "79": "toothbrush"
83
+ },
84
+ "label2id": {
85
+ "person": 0,
86
+ "bicycle": 1,
87
+ "car": 2,
88
+ "motorcycle": 3,
89
+ "airplane": 4,
90
+ "bus": 5,
91
+ "train": 6,
92
+ "truck": 7,
93
+ "boat": 8,
94
+ "traffic light": 9,
95
+ "fire hydrant": 10,
96
+ "stop sign": 11,
97
+ "parking meter": 12,
98
+ "bench": 13,
99
+ "bird": 14,
100
+ "cat": 15,
101
+ "dog": 16,
102
+ "horse": 17,
103
+ "sheep": 18,
104
+ "cow": 19,
105
+ "elephant": 20,
106
+ "bear": 21,
107
+ "zebra": 22,
108
+ "giraffe": 23,
109
+ "backpack": 24,
110
+ "umbrella": 25,
111
+ "handbag": 26,
112
+ "tie": 27,
113
+ "suitcase": 28,
114
+ "frisbee": 29,
115
+ "skis": 30,
116
+ "snowboard": 31,
117
+ "sports ball": 32,
118
+ "kite": 33,
119
+ "baseball bat": 34,
120
+ "baseball glove": 35,
121
+ "skateboard": 36,
122
+ "surfboard": 37,
123
+ "tennis racket": 38,
124
+ "bottle": 39,
125
+ "wine glass": 40,
126
+ "cup": 41,
127
+ "fork": 42,
128
+ "knife": 43,
129
+ "spoon": 44,
130
+ "bowl": 45,
131
+ "banana": 46,
132
+ "apple": 47,
133
+ "sandwich": 48,
134
+ "orange": 49,
135
+ "broccoli": 50,
136
+ "carrot": 51,
137
+ "hot dog": 52,
138
+ "pizza": 53,
139
+ "donut": 54,
140
+ "cake": 55,
141
+ "chair": 56,
142
+ "couch": 57,
143
+ "potted plant": 58,
144
+ "bed": 59,
145
+ "dining table": 60,
146
+ "toilet": 61,
147
+ "tv": 62,
148
+ "laptop": 63,
149
+ "mouse": 64,
150
+ "remote": 65,
151
+ "keyboard": 66,
152
+ "cell phone": 67,
153
+ "microwave": 68,
154
+ "oven": 69,
155
+ "toaster": 70,
156
+ "sink": 71,
157
+ "refrigerator": 72,
158
+ "book": 73,
159
+ "clock": 74,
160
+ "vase": 75,
161
+ "scissors": 76,
162
+ "teddy bear": 77,
163
+ "hair drier": 78,
164
+ "toothbrush": 79
165
+ },
166
+ "model_type": "yolov9"
167
  }
onnx/model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6a54a405cc2e58fab055d56265474bf62c82eae6ff9b5c63349f49e6da80ca49
3
- size 9141896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1622d3a267030a1d0638334e5205295ca04865e97b03220385b3755f816ee90b
3
+ size 102098161
onnx/model_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:644a00631b5acf22e9c122efb92ccecb4fc32a19c69256a42c96f48a1e438d84
3
+ size 51242537
onnx/model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:255cda648d7de9c9647737d132c8812455e4552146ed10cf5130b91c7d380f9a
3
+ size 26096497
preprocessor_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": false,
3
+ "do_pad": false,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "feature_extractor_type": "ImageFeatureExtractor",
7
+ "resample": 2,
8
+ "rescale_factor": 0.00392156862745098,
9
+ "size": {
10
+ "width": 640,
11
+ "height": 640
12
+ }
13
+ }
quantize_config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "per_channel": false,
3
+ "reduce_range": false,
4
+ "per_model_config": {
5
+ "model": {
6
+ "op_types": [
7
+ "Squeeze",
8
+ "Cast",
9
+ "Softmax",
10
+ "NonMaxSuppression",
11
+ "Reshape",
12
+ "Mul",
13
+ "Split",
14
+ "Resize",
15
+ "Gather",
16
+ "Expand",
17
+ "Concat",
18
+ "Conv",
19
+ "TopK",
20
+ "Slice",
21
+ "Range",
22
+ "Div",
23
+ "ConstantOfShape",
24
+ "GatherND",
25
+ "Transpose",
26
+ "Greater",
27
+ "Shape",
28
+ "Sub",
29
+ "Equal",
30
+ "NonZero",
31
+ "Add",
32
+ "Constant",
33
+ "MaxPool",
34
+ "Where",
35
+ "ReduceMax",
36
+ "ScatterND",
37
+ "Unsqueeze",
38
+ "ArgMax",
39
+ "Sigmoid",
40
+ "AveragePool"
41
+ ],
42
+ "weight_type": "QUInt8"
43
+ }
44
+ }
45
+ }