diff --git a/aidge_onnx/utils.py b/aidge_onnx/utils.py index 642488f494aa3a530ef8d2aeed6c59a88e1d7bfa..7a5029bae421a059dba59ce4306d7db86ca9ee6a 100644 --- a/aidge_onnx/utils.py +++ b/aidge_onnx/utils.py @@ -18,6 +18,7 @@ _MAP_NP_ONNX_TYPE = { } _MAP_ONNX_NP_TYPE = {v: k for k, v in _MAP_NP_ONNX_TYPE.items()} + def onnx_to_aidge_model_names(model: onnx.ModelProto): """ Change the name of each node of the model from onnx convention to aidge's one @@ -63,6 +64,5 @@ def onnx_to_aidge_name(name: str) -> str: Translates onnx node naming convention to aidge naming convention """ name = name.replace("/", "_").replace(".", "_") - name = name if len(name) > 0 and name[0].isdigit() else "data_" + name + name = name if (len(name) == 0 or not name[0].isdigit()) else "data_" + name return name -