OpenDevLibrary/openvino_initialization_script.py at master · chenkenanalytic/OpenDevLibrary · GitHub

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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
# -*- coding: utf-8 -*-
"""OpenVINO-ColabNotebook.ipynb
Automatically generated by Colaboratory.
Original file is located at
    https://colab.research.google.com/github/abhik-99/OpenDevLibrary/blob/master/OpenDevNotebook-CPU-modified.ipynb
"""
from subprocess import call

#Defining the Important Paths

file_name = "l_openvino_toolkit_p_2020.1.023.tgz" #change the filename if version does not match
dir_name = file_name[:-4]
install_dir = "/opt/intel/openvino/"
deployment_tools = install_dir+"deployment_tools/"
model_optimizer = install_dir+"deployment_tools/model_optimizer/"
model_zoo = deployment_tools+"open_model_zoo/"

call('wget "https://storage.googleapis.com/open_vino_public/l_openvino_toolkit_p_2020.1.023.tgz"', shell=True)

call('tar -xzf l_openvino_toolkit_p_2020.1.023.tgz', shell=True)
call('sudo -E %s/install_openvino_dependencies.sh'%(dir_name), shell=True)
call("sed -i 's/decline/accept/g' %s/silent.cfg && sed -i 's/#INTEL_SW_IMPROVEMENT/INTEL_SW_IMPROVEMENT/g' %s/silent.cfg"%(dir_name,dir_name), shell=True)
print("Installed OpenVINO Dependencies. Installing OpenVINO...")
call("sudo %s/install.sh --silent %s/silent.cfg"%(dir_name,dir_name), shell=True)
call("sudo -E %s/install_dependencies/install_openvino_dependencies.sh"%(install_dir), shell=True)
call("source %s/bin/setupvars.sh"%(install_dir), shell=True)
print("ENV Variables Set!")
# frameworks = ['tf','mxnet','onnx','kaldi','all']
# choices = dict(zip(range(1,6),frameworks))

# print("""Please enter the Choice of framework you want to work with:
# \n(Note: You should only install for the ones you would be using.
# Incase of needing to install for more than one but not all, rerun this cell and
# install the pre-requisites one by one.)
# """)

# for x in choices:
#   print(x,choices[x])

# choice = input("Please enter your choice (Default Option - 5): ")
# if len(choice) == 0:
#     choice = 5
# elif choice in '1 2 3 4'.split():
#     choice = int(choice)
# if choice>5:
#     print("You have entered an invalid choice! Please rerun the script.")

# print("Choice is",choice,":",choices[choice])
# if choice != 5:
#    pre_install = model_optimizer + "install_prerequisites/install_prerequisites.sh "+choices[choice]
#    call("sudo %s"%(pre_install), shell=True)
# elif choice == 5:
#   # for x in choices:
#   #   pre_install = model_optimizer + "install_prerequisites/install_prerequisites.sh "+choices[x]
  #   !sudo $pre_install
call("sudo %s/install_prerequisites/install_prerequisites.sh"%(model_optimizer), shell=True)
# else:
#   print("Wrong Choice! Please rerun this cell and enter the correct choice!")

call("sudo %s/demo/demo_squeezenet_download_convert_run.sh"%(deployment_tools), shell=True)

print("\n\nIntel OpenVINO Installation Fisnished!")
print("Copy and Paste the following code snippet in the next cell of your notebook")
print("#-----Start Copying from below-----")
print("")
print(f"install_dir = '{install_dir}'")
print(f"model_optimizer = '{model_optimizer}'")
print(f"deployment_tools = '{deployment_tools}'")
print(f"model_zoo = '{model_zoo}'")
print("")
print("#-----End of Copying-----")
#print("It is recommended that you save these paths to variables in the Python runtime/Jupyter Notebooks")