Ivan Tikhonov
c1608628d4
* LowLatency 2.0: transformation and unit tests * low latency 2.0: unit tests * documentation and ngraph codestyle * update CNN Interface of LowLatency transformation * fix build on Windows * fix build on Windows * investigation of a failed build on Win OS * ngraph codestyle * fix build (werrors) * New unit tests, refactoring * update functional tests for Memory * update LowLatency functional tests * extend Memory tests to cover LowLatency v2 transformation * clean up, code style * fix unit tests * update and fix unit tests, add feature to apply LLTv2 after LLTv1 * update docs, refactoring * add several gna tests to skip config * fix python api tests * update python api, rename LowLatency_v2 to LowLatency2 * deprecate LowLatency v1 * Deprecate LowLatency v1 in IE * fix wrong merge, codestyle * resolve review comments * fix python test * update skip config * apply online review notes, fix unit tests * clean up, code style * fix docs * Use debug_messages instead of exceptions in llt v2 * fix unit tests * Resolve review remarks
51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
# Copyright (C) 2021 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import argparse
|
|
|
|
from mo.utils.error import Error
|
|
from mo.utils.cli_parser import parse_transform
|
|
|
|
|
|
def get_available_transformations():
|
|
try:
|
|
from openvino.offline_transformations import ApplyLowLatencyTransformation # pylint: disable=import-error,no-name-in-module
|
|
return {
|
|
'LowLatency2': ApplyLowLatencyTransformation,
|
|
}
|
|
except Exception as e:
|
|
return {}
|
|
|
|
|
|
def apply_offline_transformations(input_model: str, framework: str, transforms: list):
|
|
# This variable is only needed by GenerateMappingFile transformation
|
|
# to produce correct mapping
|
|
extract_names = framework in ['tf', 'mxnet', 'kaldi']
|
|
|
|
from openvino.inference_engine import read_network # pylint: disable=import-error,no-name-in-module
|
|
from openvino.offline_transformations import ApplyMOCTransformations, GenerateMappingFile # pylint: disable=import-error,no-name-in-module
|
|
|
|
net = read_network(input_model + "_tmp.xml", input_model + "_tmp.bin")
|
|
|
|
available_transformations = get_available_transformations()
|
|
|
|
for name, args in transforms:
|
|
if name not in available_transformations.keys():
|
|
raise Error("Transformation {} is not available.".format(name))
|
|
|
|
available_transformations[name](net, **args)
|
|
|
|
ApplyMOCTransformations(net, False)
|
|
net.serialize(input_model + ".xml", input_model + ".bin")
|
|
path_to_mapping = input_model + ".mapping"
|
|
GenerateMappingFile(net, path_to_mapping.encode('utf-8'), extract_names)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--input_model")
|
|
parser.add_argument("--framework")
|
|
parser.add_argument("--transform")
|
|
args = parser.parse_args()
|
|
|
|
apply_offline_transformations(args.input_model, args.framework, parse_transform(args.transform)) |