Skip to content
This repository has been archived by the owner on Nov 11, 2023. It is now read-only.

Commit

Permalink
Interpolate bug fixes and support for U64, U32, and U16
Browse files Browse the repository at this point in the history
  • Loading branch information
PINTO0309 committed Jun 24, 2022
1 parent 0b7b726 commit 0c04351
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 4 deletions.
6 changes: 4 additions & 2 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ openvino2tensorflow/saved_model/

*.bin
*.xml
replace.json
*.json
!weight_replacement_config_sample.json
saved_model/
packages/
packages/
.pkl*
6 changes: 5 additions & 1 deletion openvino2tensorflow/openvino2tensorflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,9 @@ def convert(
'I32' : ['i', 4],
'I16' : ['h', 2],
'I8' : ['b', 1],
'U64' : ['Q', 8],
'U32' : ['I', 4],
'U16' : ['H', 2],
'U8' : ['B', 1],
'BOOL' : ['?', 1]
}
Expand Down Expand Up @@ -527,7 +530,8 @@ def layer_structure_print(info: dict) -> None:
layer.attrib['type'] == 'ReverseSequence' or \
layer.attrib['type'] == 'Range' or \
layer.attrib['type'] == 'Einsum' or \
layer.attrib['type'] == 'ScatterUpdate':
layer.attrib['type'] == 'ScatterUpdate' or \
layer.attrib['type'] == 'Interpolate':
concat_port_list.setdefault(to_layer, []).append(f'{from_layer}:{to_layer_port}')

for layer in layers:
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
setup(
name="openvino2tensorflow",
scripts=scripts,
version="1.31.3",
version="1.31.4",
description="This script converts the OpenVINO IR model to Tensorflow's saved_model, tflite, h5 and pb. in (NCHW) format",
long_description=long_description,
long_description_content_type="text/markdown",
Expand Down

0 comments on commit 0c04351

Please sign in to comment.