unicorn who dev commited on
Complete FireViewer Android learning SDK, source and verification records
Browse files
conversion/package/scripts/convert_onnx.py
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Fix batch/spatial dimensions explicitly before the direct FlatBuffer export."""
|
| 2 |
+
import json,subprocess,sys
|
| 3 |
+
from pathlib import Path
|
| 4 |
+
import onnx
|
| 5 |
+
|
| 6 |
+
source,out=map(Path,sys.argv[1:3])
|
| 7 |
+
spatial_size=int(sys.argv[3])
|
| 8 |
+
model=onnx.load(str(source),load_external_data=False)
|
| 9 |
+
shapes=[]
|
| 10 |
+
for inp in model.graph.input:
|
| 11 |
+
dims=[]
|
| 12 |
+
for axis,dim in enumerate(inp.type.tensor_type.shape.dim):
|
| 13 |
+
if dim.dim_value>0: dims.append(dim.dim_value)
|
| 14 |
+
elif axis==0:dims.append(1)
|
| 15 |
+
elif len(inp.type.tensor_type.shape.dim)==4 and axis in {2,3}:dims.append(spatial_size)
|
| 16 |
+
else:raise ValueError(f'Unresolved input dimension: {inp.name} axis={axis}')
|
| 17 |
+
shapes.append(inp.name+':'+','.join(map(str,dims)))
|
| 18 |
+
out.mkdir(parents=True,exist_ok=True)
|
| 19 |
+
(out/'input-shapes.json').write_text(json.dumps(shapes,indent=2)+'\n')
|
| 20 |
+
subprocess.run([str(Path(sys.executable).parent/'onnx2tf'),'-i',str(source),'-o',str(out),'-b','1','-ois',*shapes],check=True)
|
reports/fireviewer-completion.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"generated_at": "2026-09-20T00:
|
| 3 |
"repo": "fireviewer/litert-models",
|
| 4 |
"models": [
|
| 5 |
{
|
|
|
|
| 1 |
{
|
| 2 |
+
"generated_at": "2026-09-20T00:36:23.287838+00:00",
|
| 3 |
"repo": "fireviewer/litert-models",
|
| 4 |
"models": [
|
| 5 |
{
|