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

Update gna_sample with API 2.0 features (#10236)

Browse Source
This commit is contained in:
Nikita Malinin
2022-02-11 13:23:02 +03:00
committed by GitHub
parent 31501a7992
commit d758a21d6e
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tools/pot/openvino/tools/pot/api/samples/speech/utils.py
View File

@@ -1,14 +1,19 @@
# Copyright (C) 2020-2022 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
import numpy as np
from openvino.tools.pot.engines.simplified_engine import SimplifiedEngine
class ArkEngine(SimplifiedEngine):
def _fill_input(self, model, image_batch):
if 'input_names' in self.data_loader.config:
model_inputs = {n.get_node().friendly_name: n for n in model.inputs}
feed_dict = {}
for input_name in self.data_loader.config['input_names']:
feed_dict[input_name] = image_batch[0][input_name]
input_blob = model_inputs[input_name]
input_blob_name = self._get_input_any_name(input_blob)
input_blob_shape = list(input_blob.shape)
feed_dict[input_blob_name] = np.reshape(image_batch[0][input_name], input_blob_shape)
return feed_dict
raise Exception('input_names is not provided!')