def get_args(args=None):
from mo.utils.cli_parser import get_onnx_cli_parser
parser = get_onnx_cli_parser(parser=None) # setup for mo
parser.add_argument('test_dir')
parser.add_argument('--debug', '-g', action='store_true')
parser.add_argument('--force_mo', action='store_true')
parser.add_argument('--iterations', '-I', type=int, default=1)
# for inference-engine
parser.add_argument('--device',
choices=['CPU', 'GPU', 'MYRIAD'],
default='CPU')
parser.add_argument('--plugin_dir',
help='Path to a plugin folder',
default=None)
parser.add_argument(
'--cpu_extension',
help='Required for CPU custom layers. '
'MKLDNN (CPU)-targeted custom layers. Absolute path to a shared '
'library with the kernels implementations',
default=None)
parser.add_argument('--model_file', default=None)
args = parser.parse_args(args=args)
args.framework = 'onnx'
if args.debug:
args.log_level = 'DEBUG'
return args
#!/usr/bin/env python3
"""
Copyright (C) 2018-2021 Intel Corporation
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
import sys
from mo.utils.versions_checker import check_python_version
if __name__ == "__main__":
ret_code = check_python_version()
if ret_code:
sys.exit(ret_code)
from mo.main import main
from mo.utils.cli_parser import get_onnx_cli_parser
sys.exit(main(get_onnx_cli_parser(), 'onnx'))
# Copyright (C) 2021 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
import sys
from mo.utils.cli_parser import get_onnx_cli_parser
if __name__ == "__main__":
from mo.main import main
from ngraph.frontend import FrontEndManager # pylint: disable=no-name-in-module,import-error
sys.exit(main(get_onnx_cli_parser(), FrontEndManager(), 'onnx'))
