forked from openvinotoolkit/training_extensions
-
Notifications
You must be signed in to change notification settings - Fork 0
/
init_venv.sh
executable file
·45 lines (31 loc) · 1.12 KB
/
init_venv.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#!/usr/bin/env bash
work_dir=$(realpath "$(dirname $0)")
cd ${work_dir}
if [[ -e venv ]]; then
echo
echo "Virtualenv already exists. Use command to start working:"
echo "$ . venv/bin/activate"
fi
# Download mmdetection
git submodule update --init --recommend-shallow ../../external/mmdetection
# Create virtual environment
virtualenv venv -p python3 --prompt="(detection)"
path_openvino_vars="${INTEL_OPENVINO_DIR:-/opt/intel/openvino}/bin/setupvars.sh"
if [[ -e "${path_openvino_vars}" ]]; then
echo ". ${path_openvino_vars}" >> venv/bin/activate
fi
. venv/bin/activate
cat requirements.txt | xargs -n 1 -L 1 pip3 install
mo_requirements_file="${INTEL_OPENVINO_DIR:-/opt/intel/openvino}/deployment_tools/model_optimizer/requirements_onnx.txt"
if [[ -e "${mo_requirements_file}" ]]; then
pip install -qr ${mo_requirements_file}
else
echo "[WARNING] Model optimizer requirements were not installed. Please install the OpenVino toolkit to use one."
fi
cd ../../external/mmdetection/
bash compile.sh
python setup.py develop
deactivate
echo
echo "Activate a virtual environment to start working:"
echo "$ . venv/bin/activate"