2bf8d910f6
* [Docs][PyOV] update python snippets * first snippet * Fix samples debug * Fix linter * part1 * Fix speech sample * update model state snippet * add serialize * add temp dir * CPU snippets update (#134) * snippets CPU 1/6 * snippets CPU 2/6 * snippets CPU 3/6 * snippets CPU 4/6 * snippets CPU 5/6 * snippets CPU 6/6 * make module TODO: REMEMBER ABOUT EXPORTING PYTONPATH ON CIs ETC * Add static model creation in snippets for CPU * export_comp_model done * leftovers * apply comments * apply comments -- properties * small fixes * rempve debug info * return IENetwork instead of Function * apply comments * revert precision change in common snippets * update opset * [PyOV] Edit docs for the rest of plugins (#136) * modify main.py * GNA snippets * GPU snippets * AUTO snippets * MULTI snippets * HETERO snippets * Added properties * update gna * more samples * Update docs/OV_Runtime_UG/model_state_intro.md * Update docs/OV_Runtime_UG/model_state_intro.md * attempt1 fix ci * new approach to test * temporary remove some files from run * revert cmake changes * fix ci * fix snippet * fix py_exclusive snippet * fix preprocessing snippet * clean-up main * remove numpy installation in gha * check for GPU * add logger * iexclude main * main update * temp * Temp2 * Temp2 * temp * Revert temp * add property execution devices * hide output from samples --------- Co-authored-by: p-wysocki <przemyslaw.wysocki@intel.com> Co-authored-by: Jan Iwaszkiewicz <jan.iwaszkiewicz@intel.com> Co-authored-by: Karol Blaszczak <karol.blaszczak@intel.com>
87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
# Copyright (C) 2018-2023 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
import tempfile
|
|
import numpy as np
|
|
from sys import platform
|
|
|
|
import openvino as ov
|
|
import openvino.runtime.opset12 as ops
|
|
|
|
import ngraph as ng
|
|
from ngraph.impl import Function
|
|
import openvino.inference_engine as ie
|
|
|
|
|
|
def get_dynamic_model():
|
|
param = ops.parameter(ov.PartialShape([-1, -1]), name="input")
|
|
return ov.Model(ops.relu(param), [param])
|
|
|
|
|
|
def get_model(input_shape = None, input_dtype=np.float32) -> ov.Model:
|
|
if input_shape is None:
|
|
input_shape = [1, 3, 32, 32]
|
|
param = ops.parameter(input_shape, input_dtype, name="input")
|
|
relu = ops.relu(param, name="relu")
|
|
relu.output(0).get_tensor().set_names({"result"})
|
|
model = ov.Model([relu], [param], "test_model")
|
|
|
|
assert model is not None
|
|
return model
|
|
|
|
|
|
def get_advanced_model():
|
|
data = ops.parameter(ov.Shape([1, 3, 64, 64]), ov.Type.f32, name="input")
|
|
data1 = ops.parameter(ov.Shape([1, 3, 64, 64]), ov.Type.f32, name="input2")
|
|
axis_const = ops.constant(1, dtype=ov.Type.i64)
|
|
split = ops.split(data, axis_const, 3)
|
|
relu = ops.relu(split.output(1))
|
|
relu.output(0).get_tensor().set_names({"result"})
|
|
return ov.Model([split.output(0), relu.output(0), split.output(2)], [data, data1], "model")
|
|
|
|
|
|
def get_ngraph_model(input_shape = None, input_dtype=np.float32):
|
|
if input_shape is None:
|
|
input_shape = [1, 3, 32, 32]
|
|
param = ng.opset11.parameter(input_shape, input_dtype, name="data")
|
|
relu = ng.opset11.relu(param, name="relu")
|
|
func = Function([relu], [param], "test_model")
|
|
caps = ng.Function.to_capsule(func)
|
|
net = ie.IENetwork(caps)
|
|
|
|
assert net is not None
|
|
return net
|
|
|
|
|
|
def get_image(shape = (1, 3, 32, 32), dtype = "float32"):
|
|
np.random.seed(42)
|
|
return np.random.rand(*shape).astype(dtype)
|
|
|
|
|
|
def get_path_to_extension_library():
|
|
library_path=""
|
|
if platform == "win32":
|
|
library_path="openvino_template_extension.dll"
|
|
else:
|
|
library_path="libopenvino_template_extension.so"
|
|
return library_path
|
|
|
|
|
|
def get_path_to_model(input_shape = None, is_old_api=False):
|
|
if input_shape is None:
|
|
input_shape = [1, 3, 32, 32]
|
|
path_to_xml = tempfile.NamedTemporaryFile(suffix="_model.xml").name
|
|
if is_old_api:
|
|
net = get_ngraph_model(input_shape, input_dtype=np.int64)
|
|
net.serialize(path_to_xml)
|
|
else:
|
|
model = get_model(input_shape)
|
|
ov.serialize(model, path_to_xml)
|
|
return path_to_xml
|
|
|
|
|
|
def get_temp_dir():
|
|
temp_dir = tempfile.TemporaryDirectory()
|
|
return temp_dir
|