leap-model-parser 0.1.159__py3-none-any.whl → 0.1.161__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- leap_model_parser/model_parser.py +3 -0
- {leap_model_parser-0.1.159.dist-info → leap_model_parser-0.1.161.dist-info}/METADATA +2 -2
- {leap_model_parser-0.1.159.dist-info → leap_model_parser-0.1.161.dist-info}/RECORD +6 -6
- {leap_model_parser-0.1.159.dist-info → leap_model_parser-0.1.161.dist-info}/LICENSE +0 -0
- {leap_model_parser-0.1.159.dist-info → leap_model_parser-0.1.161.dist-info}/WHEEL +0 -0
- {leap_model_parser-0.1.159.dist-info → leap_model_parser-0.1.161.dist-info}/entry_points.txt +0 -0
|
@@ -13,6 +13,7 @@ from keras_data_format_converter import convert_channels_first_to_last # type:
|
|
|
13
13
|
from leap_model_rebuilder import rebuild_model # type: ignore
|
|
14
14
|
from onnx2kerastl import onnx_to_keras # type: ignore
|
|
15
15
|
from onnx2kerastl.converter import ConvertedResponse # type: ignore
|
|
16
|
+
from onnx2kerastl.customonnxlayer import onnx_custom_layers
|
|
16
17
|
from onnx2kerastl.exceptions import OnnxUnsupported # type: ignore
|
|
17
18
|
from tensorflow.keras.models import load_model # type: ignore
|
|
18
19
|
|
|
@@ -37,6 +38,8 @@ class ModelParser:
|
|
|
37
38
|
if custom_layers is None:
|
|
38
39
|
self.custom_layers = {}
|
|
39
40
|
|
|
41
|
+
self.custom_layers = {**self.custom_layers, **onnx_custom_layers}
|
|
42
|
+
|
|
40
43
|
self._model_types_converter = {
|
|
41
44
|
ImportModelTypeEnum.JSON_TF2.value: self.convert_json_model,
|
|
42
45
|
ImportModelTypeEnum.H5_TF2.value: self.convert_h5_model,
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: leap-model-parser
|
|
3
|
-
Version: 0.1.
|
|
3
|
+
Version: 0.1.161
|
|
4
4
|
Summary:
|
|
5
5
|
Home-page: https://github.com/tensorleap/leap-model-parser
|
|
6
6
|
License: MIT
|
|
@@ -16,7 +16,7 @@ Requires-Dist: keras-data-format-converter (==0.1.17)
|
|
|
16
16
|
Requires-Dist: leap-model-rebuilder (==0.1.5)
|
|
17
17
|
Requires-Dist: numpy (>=1.22.3,<2.0.0)
|
|
18
18
|
Requires-Dist: onnx (==1.10.1)
|
|
19
|
-
Requires-Dist: onnx2kerastl (==0.0.
|
|
19
|
+
Requires-Dist: onnx2kerastl (==0.0.148)
|
|
20
20
|
Requires-Dist: tensorflow (==2.11.0) ; platform_machine == "x86_64"
|
|
21
21
|
Requires-Dist: tensorflow-io-gcs-filesystem (==0.30.0)
|
|
22
22
|
Requires-Dist: tensorflow-macos (==2.11.0) ; platform_machine == "arm64"
|
|
@@ -6,7 +6,7 @@ leap_model_parser/contract/importmodelresponse.py,sha256=GlvnKS8xrebU2Sj0dxqtEhA
|
|
|
6
6
|
leap_model_parser/contract/nodedata.py,sha256=1_ML0nzp3QUZ0_9mGSLhfO4_hqjYMwi0DWLwymUnWEs,43326
|
|
7
7
|
leap_model_parser/contract/ui_components.json,sha256=lhKWOQjvGCxQakDcNHq7aBmORUf67ZodZcCtJ_TIcsI,416303
|
|
8
8
|
leap_model_parser/keras_json_model_import.py,sha256=f3ISFoiXPgxzIXAxvbiIKyxcvq4Z-Xv8X0O2I9BQA5k,16546
|
|
9
|
-
leap_model_parser/model_parser.py,sha256=
|
|
9
|
+
leap_model_parser/model_parser.py,sha256=2GwzI_Quu8Z3rR50B1vyzVZL4ckvZ5D1-eZAlb_RSkA,7246
|
|
10
10
|
leap_model_parser/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
11
11
|
leap_model_parser/utils/layerpedia/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
12
12
|
leap_model_parser/utils/layerpedia/layerpedia.py,sha256=1syubfXBTB630TVkgcQ-Ge7Qe9Zbr6EtZRreuqCJnQ8,9292
|
|
@@ -16,8 +16,8 @@ leap_model_parser/utils/uicomponents/__init__.py,sha256=ShNUN5NLzaxOQ5ByNUUWJG6v
|
|
|
16
16
|
leap_model_parser/utils/uicomponents/generatenodedata.py,sha256=LRaPlO5jJ9pUtkvLG9QaGBRnu860wNH_OkBM6UKLGGQ,3035
|
|
17
17
|
leap_model_parser/utils/uicomponents/tensorflowinscpection.py,sha256=ym613z9iQKPDBpr0RYD35bTABdm1L-Ez86G47BYT7qw,6775
|
|
18
18
|
leap_model_parser/utils/uicomponents/ui_components_config.yaml,sha256=nzhcSR-00erM-D09CczOmfF4a6w9kZUgDjDksHtLsUo,19604
|
|
19
|
-
leap_model_parser-0.1.
|
|
20
|
-
leap_model_parser-0.1.
|
|
21
|
-
leap_model_parser-0.1.
|
|
22
|
-
leap_model_parser-0.1.
|
|
23
|
-
leap_model_parser-0.1.
|
|
19
|
+
leap_model_parser-0.1.161.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
|
|
20
|
+
leap_model_parser-0.1.161.dist-info/METADATA,sha256=xA7CQSP-NsgO_USlULM63yJq5FpKVjTp4yp9tY5FAX4,1065
|
|
21
|
+
leap_model_parser-0.1.161.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
22
|
+
leap_model_parser-0.1.161.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
|
|
23
|
+
leap_model_parser-0.1.161.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
{leap_model_parser-0.1.159.dist-info → leap_model_parser-0.1.161.dist-info}/entry_points.txt
RENAMED
|
File without changes
|