Skip to content
Snippets Groups Projects
Commit e6fdc573 authored by Grégoire Kubler's avatar Grégoire Kubler Committed by Cyril Moineau
Browse files

chore : renamed function

parent 4ec6826d
No related branches found
No related tags found
2 merge requests!390.2.1,!33feat/script_compare_layers_aidge_onnx
...@@ -15,7 +15,7 @@ import colorama ...@@ -15,7 +15,7 @@ import colorama
from onnx import numpy_helper from onnx import numpy_helper
import onnx import onnx
from .node_import import ONNX_NODE_CONVERTER_, generic from .node_import import ONNX_NODE_CONVERTER_, generic
from .utils import onnx_to_aidge_convert_model_names from .utils import onnx_to_aidge_model_names
def load_onnx(filename: str, verbose: bool = False): def load_onnx(filename: str, verbose: bool = False):
"""Load an ONNX file and convert it into a :py:class:`aidge_core.GraphView`. """Load an ONNX file and convert it into a :py:class:`aidge_core.GraphView`.
...@@ -107,7 +107,7 @@ def _load_onnx2graphview(model:onnx.ModelProto, verbose:bool = False): ...@@ -107,7 +107,7 @@ def _load_onnx2graphview(model:onnx.ModelProto, verbose:bool = False):
# Clean model if some issues in the model # Clean model if some issues in the model
# might affect Aidge in the next steps # might affect Aidge in the next steps
model = onnx_to_aidge_convert_model_names(model) model = onnx_to_aidge_model_names(model)
if verbose : print(f"\nGetting Initializers\n====================") if verbose : print(f"\nGetting Initializers\n====================")
# Get the initializers # Get the initializers
for i in model.graph.initializer: for i in model.graph.initializer:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment