67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
# Copyright (C) 2021 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import logging as log
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from mo.utils.versions_checker import check_python_version # pylint: disable=no-name-in-module
|
|
|
|
|
|
def log_ie_not_found():
|
|
log.error("Could not find the Inference Engine or nGraph Python API.\n"
|
|
"Consider building the Inference Engine and nGraph Python APIs"
|
|
" from sources or try to install OpenVINO (TM) Toolkit using \"install_prerequisites.{}\""
|
|
.format("bat" if sys.platform == "windows" else "sh"))
|
|
|
|
|
|
def setup_env():
|
|
ret_code = check_python_version()
|
|
if ret_code:
|
|
sys.exit(ret_code)
|
|
|
|
from mo.utils.find_ie_version import find_ie_version
|
|
|
|
ie_found = True
|
|
try:
|
|
ie_found = find_ie_version(silent=True)
|
|
except Exception:
|
|
ie_found = False
|
|
|
|
if not ie_found:
|
|
log_ie_not_found()
|
|
sys.exit(1)
|
|
|
|
mo_root_path = os.path.join(os.path.dirname(__file__), os.pardir)
|
|
|
|
python_path_key = 'PYTHONPATH'
|
|
if python_path_key not in os.environ:
|
|
os.environ[python_path_key] = mo_root_path
|
|
else:
|
|
os.environ[python_path_key] = os.pathsep.join([os.environ[python_path_key], mo_root_path])
|
|
return True
|
|
|
|
|
|
def subprocess_main(framework=None):
|
|
"""
|
|
Please keep this file compatible with python2 in order to check user python version.
|
|
|
|
This function checks that Inference Engine Python API available and working as expected
|
|
and then in sub-process it executes main_<fw>.py files. Due to some OSs specifics we can't
|
|
just add paths to Python modules and libraries into current env. So to make Inference Engine
|
|
Python API to be available inside MO we need to use subprocess with new env.
|
|
"""
|
|
setup_env()
|
|
|
|
path_to_main = os.path.join(os.path.realpath(os.path.dirname(__file__)),
|
|
'main_{}.py'.format(framework) if framework else 'main.py')
|
|
|
|
# python2 compatible code. Do not remove.
|
|
args = [sys.executable, path_to_main]
|
|
|
|
for arg in sys.argv[1:]:
|
|
args.append(arg)
|
|
status = subprocess.run(args, env=os.environ)
|
|
sys.exit(status.returncode)
|