wip

README.md

@@ -8,4 +8,9 @@ base_model:
 
 ONNX model converted and optimized from `BirdNET_GLOBAL_6K_V2.4_Model_FP32.tflite`.
 
+Files:
+
+- `model.onnx`: Initial model converted with tf2onnx and edited using NVIDIA Nsight DL Designer
+- `birdnet.onnx`: Model further optimized with the `scripts/optimize.py` script. Recommended
+
 Source: https://github.com/birdnet-team/BirdNET-Analyzer

scripts/optimize.py

@@ -0,0 +1,17 @@
+import onnxscript
+import onnx_ir as ir
+import onnx_ir.passes.common
+
+
+class ReplaceDftWithMatMulRule(onnxscript.rewriter.RewriteRuleClassBase):
+    
+
+model = ir.load("model.onnx")
+onnxscript.optimizer.optimize(
+    model, input_size_limit=1024 * 1024 * 1024, output_size_limit=1024 * 1024 * 1024
+)
+
+onnx_ir.passes.common.ClearMetadataAndDocStringPass()(model)
+model.ir_version = 10
+
+ir.save(model, "birdnet.onnx")