IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated model_creation_sample.py (#19989)

Browse Source 
Removed usage of ngraph in the sample
This commit is contained in:
Georgy Krivoruchko
2023-09-21 15:05:24 -07:00
committed by GitHub
parent 058b45e608
commit 10c3b60aac
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
samples/python/model_creation_sample/model_creation_sample.py
View File

@@ -14,8 +14,8 @@ from openvino.runtime import op, opset1, opset8
from data import digits
def create_ngraph_function(model_path: str) -> ov.Model:
"""Create a model on the fly from the source code using ngraph."""
def create_model(model_path: str) -> ov.Model:
"""Create a model on the fly from the source code using openvino."""
def shape_and_length(shape: list) -> typing.Tuple[list, int]:
length = reduce(lambda x, y: x * y, shape)
@@ -136,8 +136,8 @@ def main():
log.info('Creating OpenVINO Runtime Core')
# ---------------------------Step 2. Read a model in OpenVINO Intermediate Representation------------------------------
log.info(f'Loading the model using ngraph function with weights from {model_path}')
model = create_ngraph_function(model_path)
log.info(f'Loading the model using openvino with weights from {model_path}')
model = create_model(model_path)
# ---------------------------Step 3. Apply preprocessing----------------------------------------------------------
# Get names of input and output blobs
ppp = ov.preprocess.PrePostProcessor(model)