Hugging Face's logo

software-mansion
/
react-native-executorch-yolo26-seg

Image Segmentation
ExecuTorch
Model card Files
xet
 Community
msluszniak commited on
Commit
b3045ff
·
verified ·
1 Parent(s): 70dfc09

Add coreml fp16 export (ExecuTorch 1.4.1)

Browse files
Files changed (1) hide show
  1. m/coreml/config.json +163 -0
m/coreml/config.json ADDED
@@ -0,0 +1,163 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "yolo26_seg",
4
+ "family": "yolo",
5
+ "capabilities": [
6
+ "object-detection",
7
+ "instance-segmentation"
8
+ ],
9
+ "backend": "coreml",
10
+ "license": "agpl-3.0",
11
+ "variants": [
12
+ {
13
+ "file": "yolo26_seg_m_384_coreml_fp16.pte",
14
+ "precision": "fp16",
15
+ "quantized": false,
16
+ "default": true,
17
+ "methods": {
18
+ "forward": {
19
+ "inputs": [
20
+ {
21
+ "shape": [
22
+ 1,
23
+ 3,
24
+ 384,
25
+ 384
26
+ ],
27
+ "dtype": "float32"
28
+ }
29
+ ],
30
+ "outputs": [
31
+ {
32
+ "shape": [
33
+ 300,
34
+ 4
35
+ ],
36
+ "dtype": "float32"
37
+ },
38
+ {
39
+ "shape": [
40
+ 300
41
+ ],
42
+ "dtype": "float32"
43
+ },
44
+ {
45
+ "shape": [
46
+ 300
47
+ ],
48
+ "dtype": "float32"
49
+ },
50
+ {
51
+ "shape": [
52
+ 300,
53
+ 96,
54
+ 96
55
+ ],
56
+ "dtype": "float32"
57
+ }
58
+ ]
59
+ }
60
+ }
61
+ },
62
+ {
63
+ "file": "yolo26_seg_m_512_coreml_fp16.pte",
64
+ "precision": "fp16",
65
+ "quantized": false,
66
+ "default": false,
67
+ "methods": {
68
+ "forward": {
69
+ "inputs": [
70
+ {
71
+ "shape": [
72
+ 1,
73
+ 3,
74
+ 512,
75
+ 512
76
+ ],
77
+ "dtype": "float32"
78
+ }
79
+ ],
80
+ "outputs": [
81
+ {
82
+ "shape": [
83
+ 300,
84
+ 4
85
+ ],
86
+ "dtype": "float32"
87
+ },
88
+ {
89
+ "shape": [
90
+ 300
91
+ ],
92
+ "dtype": "float32"
93
+ },
94
+ {
95
+ "shape": [
96
+ 300
97
+ ],
98
+ "dtype": "float32"
99
+ },
100
+ {
101
+ "shape": [
102
+ 300,
103
+ 128,
104
+ 128
105
+ ],
106
+ "dtype": "float32"
107
+ }
108
+ ]
109
+ }
110
+ }
111
+ },
112
+ {
113
+ "file": "yolo26_seg_m_640_coreml_fp16.pte",
114
+ "precision": "fp16",
115
+ "quantized": false,
116
+ "default": false,
117
+ "methods": {
118
+ "forward": {
119
+ "inputs": [
120
+ {
121
+ "shape": [
122
+ 1,
123
+ 3,
124
+ 640,
125
+ 640
126
+ ],
127
+ "dtype": "float32"
128
+ }
129
+ ],
130
+ "outputs": [
131
+ {
132
+ "shape": [
133
+ 300,
134
+ 4
135
+ ],
136
+ "dtype": "float32"
137
+ },
138
+ {
139
+ "shape": [
140
+ 300
141
+ ],
142
+ "dtype": "float32"
143
+ },
144
+ {
145
+ "shape": [
146
+ 300
147
+ ],
148
+ "dtype": "float32"
149
+ },
150
+ {
151
+ "shape": [
152
+ 300,
153
+ 160,
154
+ 160
155
+ ],
156
+ "dtype": "float32"
157
+ }
158
+ ]
159
+ }
160
+ }
161
+ }
162
+ ]
163
+ }