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>
81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
# Copyright (C) 2018-2023 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
import openvino as ov
|
|
import openvino.runtime.opset12 as ops
|
|
|
|
# ! [ov:layout:simple]
|
|
from openvino import Layout
|
|
layout = Layout('NCHW')
|
|
# ! [ov:layout:simple]
|
|
# ! [ov:layout:complex]
|
|
# Each dimension has name separated by comma
|
|
# Layout is wrapped with square brackets
|
|
layout = Layout('[time,temperature,humidity]')
|
|
# ! [ov:layout:complex]
|
|
# ! [ov:layout:partially_defined]
|
|
# First dimension is batch, 4th is 'channels'.
|
|
# Others are not important for us
|
|
layout = Layout('N??C')
|
|
|
|
# Or the same using advanced syntax
|
|
layout = Layout('[n,?,?,c]')
|
|
# ! [ov:layout:partially_defined]
|
|
# ! [ov:layout:dynamic]
|
|
# First dimension is 'batch' others are whatever
|
|
layout = Layout('N...')
|
|
|
|
# Second dimension is 'channels' others are whatever
|
|
layout = Layout('?C...')
|
|
|
|
# Last dimension is 'channels' others are whatever
|
|
layout = Layout('...C')
|
|
# ! [ov:layout:dynamic]
|
|
|
|
# ! [ov:layout:predefined]
|
|
from openvino.runtime import layout_helpers
|
|
# returns 0 for batch
|
|
layout_helpers.batch_idx(Layout('NCDHW'))
|
|
|
|
# returns 1 for channels
|
|
layout_helpers.channels_idx(Layout('NCDHW'))
|
|
|
|
# returns 2 for depth
|
|
layout_helpers.depth_idx(Layout('NCDHW'))
|
|
|
|
# returns -2 for height
|
|
layout_helpers.height_idx(Layout('...HW'))
|
|
|
|
# returns -1 for width
|
|
layout_helpers.width_idx(Layout('...HW'))
|
|
# ! [ov:layout:predefined]
|
|
|
|
# ! [ov:layout:dump]
|
|
layout = Layout('NCHW')
|
|
print(layout) # prints [N,C,H,W]
|
|
# ! [ov:layout:dump]
|
|
|
|
|
|
def create_simple_model():
|
|
# This example shows how to create ov::Function
|
|
#
|
|
# Parameter--->Multiply--->Add--->Result
|
|
# Constant---' /
|
|
# Constant---'
|
|
data = ops.parameter([3, 1, 2], ov.Type.f32, name="input_tensor_name")
|
|
mul_constant = ops.constant([1.5], ov.Type.f32)
|
|
mul = ops.multiply(data, mul_constant)
|
|
add_constant = ops.constant([0.5], ov.Type.f32)
|
|
add = ops.add(mul, add_constant)
|
|
res = ops.result(add)
|
|
return ov.Model([res], [data], "model")
|
|
|
|
model = create_simple_model()
|
|
|
|
# ! [ov:layout:get_from_model]
|
|
# Get layout for model input
|
|
layout = layout_helpers.get_layout(model.input("input_tensor_name"))
|
|
# Get layout for model with single output
|
|
layout = layout_helpers.get_layout(model.output())
|
|
# ! [ov:layout:get_from_model]
|