unicorn who dev commited on
Commit
330e909
·
verified ·
1 Parent(s): dee7f43

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:35:11.670200+00:00",
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
  {