From 6aacb0d9206720ca124955c61082f29b8b2d62a4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20W=C3=BCnsch?= Date: Fri, 9 Jan 2026 10:38:45 +0100 Subject: [PATCH 1/2] [add] Added onnx simplifier --- .../machine_learning_model/onnx_network.py | 17 ++++++++++++++++- requirements.txt | 1 + 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/ada_verona/database/machine_learning_model/onnx_network.py b/ada_verona/database/machine_learning_model/onnx_network.py index 7238bbde..0cd589aa 100644 --- a/ada_verona/database/machine_learning_model/onnx_network.py +++ b/ada_verona/database/machine_learning_model/onnx_network.py @@ -19,6 +19,7 @@ import onnx import torch from onnx2torch import convert +from onnxsim import simplify from ada_verona.database.machine_learning_model.network import Network from ada_verona.database.machine_learning_model.torch_model_wrapper import TorchModelWrapper @@ -102,7 +103,21 @@ def load_pytorch_model(self) -> torch.nn.Module: device = torch.device("cuda" if torch.cuda.is_available() else "cpu") torch_model_wrapper = self.torch_model_wrapper if torch_model_wrapper is None: - torch_model = convert(self.path).to(device) + onnx_model = self.load_onnx_model() + # Simplify model + try: + model_simp, check = simplify(onnx_model) + if not check: + print(f"ONNX-simplifier validation failed for {self.name}, using original.") + model_to_convert = onnx_model + else: + model_to_convert = model_simp + except Exception as e: + print(f"Simplification failed ({e}). Attempting to convert original model.") + model_to_convert = onnx_model + + torch_model = convert(model_to_convert).to(device) + torch_model_wrapper = TorchModelWrapper(torch_model, self.get_input_shape()) self.torch_model_wrapper = torch_model_wrapper diff --git a/requirements.txt b/requirements.txt index 9fb2e33d..a58f9116 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,6 +6,7 @@ numpy>=1.24.3 onnx>=1.14.0 onnxruntime>=1.14.1 onnx2torch>=1.5.14 +onnx-simplifier>=0.4.0 pandas>=2.0.1 PyYAML>=6.0.1 result>=0.9.0 From dda25c3bb599b37160f84c8d27dfb7c180a0311e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20W=C3=BCnsch?= Date: Fri, 9 Jan 2026 10:49:01 +0100 Subject: [PATCH 2/2] [mod] Change install requirements --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index a58f9116..6f840fe5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,7 @@ numpy>=1.24.3 onnx>=1.14.0 onnxruntime>=1.14.1 onnx2torch>=1.5.14 -onnx-simplifier>=0.4.0 +onnxsim>=0.4.0 pandas>=2.0.1 PyYAML>=6.0.1 result>=0.9.0