scripts/setupvars/setupvars.sh

   1 #!/bin/bash
   2
   3 # Copyright (c) 2018-2020 Intel Corporation
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #      http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16
  17 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
  18 BASE_DIR="$( dirname "$SCRIPT_DIR" )"
  19
  20 INSTALLDIR="${BASE_DIR}"
  21
  22
  23 export INTEL_OPENVINO_DIR="$INSTALLDIR"
  24 export INTEL_CVSDK_DIR="$INTEL_OPENVINO_DIR"
  25
  26 # parse command line options
  27 while [[ $# -gt 0 ]]
  28 do
  29 key="$1"
  30 case $key in
  31     -pyver)
  32     python_version=$2
  33     echo python_version = "${python_version}"
  34     shift
  35     ;;
  36     *)
  37     # unknown option
  38     ;;
  39 esac
  40 shift
  41 done
  42
  43 if [ -e $INSTALLDIR/deployment_tools/inference_engine ]; then
  44     export InferenceEngine_DIR=$INTEL_OPENVINO_DIR/deployment_tools/inference_engine/share
  45     system_type=$(\ls $INTEL_OPENVINO_DIR/deployment_tools/inference_engine/lib/)
  46     IE_PLUGINS_PATH=$INTEL_OPENVINO_DIR/deployment_tools/inference_engine/lib/$system_type
  47
  48     if [[ -e ${IE_PLUGINS_PATH}/arch_descriptions ]]; then
  49         export ARCH_ROOT_DIR=${IE_PLUGINS_PATH}/arch_descriptions
  50     fi
  51
  52     export HDDL_INSTALL_DIR=$INSTALLDIR/deployment_tools/inference_engine/external/hddl
  53     if [[ "$OSTYPE" == "darwin"* ]]; then
  54         export DYLD_LIBRARY_PATH=$INSTALLDIR/deployment_tools/inference_engine/external/mkltiny_mac/lib:$INSTALLDIR/deployment_tools/inference_engine/external/tbb/lib:${IE_PLUGINS_PATH}${DYLD_LIBRARY_PATH:+:DYLD_LIBRARY_PATH}
  55         export LD_LIBRARY_PATH=$INSTALLDIR/deployment_tools/inference_engine/external/mkltiny_mac/lib:$INSTALLDIR/deployment_tools/inference_engine/external/tbb/lib:${IE_PLUGINS_PATH}${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
  56     else
  57         export LD_LIBRARY_PATH=$HDDL_INSTALL_DIR/lib:$INSTALLDIR/deployment_tools/inference_engine/external/gna/lib:$INSTALLDIR/deployment_tools/inference_engine/external/mkltiny_lnx/lib:$INSTALLDIR/deployment_tools/inference_engine/external/tbb/lib:${IE_PLUGINS_PATH}${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
  58     fi
  59
  60     export KMB_INSTALL_DIR=$INSTALLDIR/deployment_tools/inference_engine/external/hddl_unite
  61     export LD_LIBRARY_PATH=$KMB_INSTALL_DIR/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
  62 fi
  63
  64 if [ -e $INSTALLDIR/deployment_tools/ngraph ]; then
  65     export LD_LIBRARY_PATH=$INSTALLDIR/deployment_tools/ngraph/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
  66     export ngraph_DIR=$INSTALLDIR/deployment_tools/ngraph/cmake
  67 fi
  68
  69 if [ -e "$INSTALLDIR/opencv" ]; then
  70     if [ -f "$INSTALLDIR/opencv/setupvars.sh" ]; then
  71         source "$INSTALLDIR/opencv/setupvars.sh"
  72     else
  73         export OpenCV_DIR="$INSTALLDIR/opencv/share/OpenCV"
  74         export LD_LIBRARY_PATH="$INSTALLDIR/opencv/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"
  75         export LD_LIBRARY_PATH="$INSTALLDIR/opencv/share/OpenCV/3rdparty/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"
  76     fi
  77 fi
  78
  79
  80 if [ -f "$INTEL_OPENVINO_DIR/data_processing/dl_streamer/bin/setupvars.sh" ]; then
  81     source "$INTEL_OPENVINO_DIR/data_processing/dl_streamer/bin/setupvars.sh"
  82 fi
  83
  84 export PATH="$INTEL_OPENVINO_DIR/deployment_tools/model_optimizer${PATH:+:$PATH}"
  85 export PYTHONPATH="$INTEL_OPENVINO_DIR/deployment_tools/model_optimizer${PYTHONPATH:+:$PYTHONPATH}"
  86
  87
  88 if [ -e $INTEL_OPENVINO_DIR/deployment_tools/open_model_zoo/tools/accuracy_checker ]; then
  89     export PYTHONPATH="$INTEL_OPENVINO_DIR/deployment_tools/open_model_zoo/tools/accuracy_checker:$PYTHONPATH"
  90 fi
  91
  92 if [ -e $INTEL_OPENVINO_DIR/deployment_tools/tools/post_training_optimization_toolkit ]; then
  93     export PYTHONPATH="$INTEL_OPENVINO_DIR/deployment_tools/tools/post_training_optimization_toolkit:$PYTHONPATH"
  94 fi
  95
  96 if [ -z "$python_version" ]; then
  97     python_version=$(python3 -c 'import sys; print(str(sys.version_info[0])+"."+str(sys.version_info[1]))')
  98 fi
  99
 100 OS_NAME=""
 101 if command -v lsb_release >/dev/null 2>&1; then
 102     OS_NAME=$(lsb_release -i -s)
 103 fi
 104
 105 python_bitness=$(python3 -c 'import sys; print(64 if sys.maxsize > 2**32 else 32)')
 106 if [ "$python_bitness" != "" ] && [ "$python_bitness" != "64" ] && [ "$OS_NAME" != "Raspbian" ]; then
 107     echo "[setupvars.sh] 64 bitness for Python" $python_version "is requred"
 108 fi
 109
 110 MINIMUM_REQUIRED_PYTHON_VERSION="3.6"
 111 if [[ ! -z "$python_version" && "$(printf '%s\n' "$python_version" "$MINIMUM_REQUIRED_PYTHON_VERSION" | sort -V | head -n 1)" != "$MINIMUM_REQUIRED_PYTHON_VERSION" ]]; then
 112     echo "[setupvars.sh] Unsupported Python version. Please install Python 3.6 or higher (64-bit) from https://www.python.org/downloads/"
 113 fi
 114
 115 if [ ! -z "$python_version" ]; then
 116     # add path to OpenCV API for Python 3.x
 117     export PYTHONPATH="$INTEL_OPENVINO_DIR/python/python3:$PYTHONPATH"
 118     if [[ -e $INTEL_OPENVINO_DIR/python/python$python_version ]]; then
 119         # add path to Inference Engine Python API
 120         export PYTHONPATH="$INTEL_OPENVINO_DIR/python/python$python_version:$PYTHONPATH"
 121     fi
 122 fi
 123
 124 echo "[setupvars.sh] OpenVINO environment initialized"