-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
119 lines (91 loc) · 4 KB
/
setup.py
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
import os
from gitclone import gitclone
from constants import API_MODEL_PATH, PROTOC_MAC_URL, PROTOC_MAC_ZIP, PROTOC_PATH, PROTOC_WIN_URL, PROTOC_WIN_ZIP
import shutil
import wget
from load import load_pretrained_model
'''------------------------------------------------------------------------------
api_model
Description: Download the api model
------------------------------------------------------------------------------'''
def api_model():
if os.path.exists(API_MODEL_PATH):
print('Tensorflow API Model already downloaded')
return
os.mkdir(API_MODEL_PATH)
gitclone('https://github.com/tensorflow/models', API_MODEL_PATH)
'''------------------------------------------------------------------------------
protoc
Description: Download and run protoc
------------------------------------------------------------------------------'''
def protoc():
if os.path.exists(PROTOC_PATH):
shutil.rmtree(PROTOC_PATH)
os.mkdir(PROTOC_PATH)
if os.name == 'posix':
# download protoc for mac
os.system('wget {}'.format(PROTOC_MAC_URL))
shutil.unpack_archive(PROTOC_MAC_ZIP, PROTOC_PATH)
os.remove(PROTOC_MAC_ZIP)
# permissions
os.system('chmod +x protoc/bin/protoc')
# add protoc to path
os.environ['PATH'] = os.path.join(os.getcwd(), 'protoc', 'bin') + ':' + os.environ['PATH']
# run
os.system('cd {}/models/research && protoc object_detection/protos/*.proto --python_out=.'.format(os.getcwd()))
if os.name == 'nt':
# download protoc for windows
wget.download(PROTOC_WIN_URL)
shutil.unpack_archive(PROTOC_WIN_ZIP, PROTOC_PATH)
os.remove(PROTOC_WIN_ZIP)
# add protoc to path
os.environ['PATH'] += os.pathsep + os.path.abspath(os.path.join(PROTOC_PATH, 'bin'))
# run
os.system('cd {}/models/research && protoc object_detection/protos/*.proto --python_out=.'.format(os.getcwd()))
'''------------------------------------------------------------------------------
installTFDeps
Description: Install tensorflow dependencies
------------------------------------------------------------------------------'''
def installTFDeps():
if not os.path.exists('models'):
raise Exception('Models from Tensorflow Garden must be downloaded first')
currentdir = os.getcwd()
if os.name == 'posix':
os.chdir(os.path.join(currentdir, 'models', 'research'))
shutil.copy(os.path.join('object_detection', 'packages', 'tf2', 'setup.py'), '.')
os.system('python -m pip install .')
if os.name == 'nt':
# prevent no module
os.system('pip install tensorflow --upgrade')
os.system('pip uninstall protobuf matplotlib -y')
os.system('pip install protobuf matplotlib==3.2')
os.system('pip install Pillow')
os.system('pip install pyyaml')
os.chdir(os.path.join(currentdir, 'models', 'research'))
shutil.copy(os.path.join('object_detection', 'packages', 'tf2', 'setup.py'), 'setup.py')
os.system('python setup.py build')
os.system('python setup.py install')
os.system('cd slim && pip install -e .')
os.chdir(currentdir) # reset
'''------------------------------------------------------------------------------
verify
Description: Verify the install of Tensorflow
------------------------------------------------------------------------------'''
def verify():
verification_script = os.path.join(API_MODEL_PATH, 'research', 'object_detection', 'builders', 'model_builder_tf2_test.py')
os.system('python {}'.format(verification_script))
'''------------------------------------------------------------------------------
setup
Description: Aggregate all functions above
------------------------------------------------------------------------------'''
def setup():
# Download TF Models
api_model()
# Load pretrained model
load_pretrained_model()
# Install and run protoc
protoc()
installTFDeps()
# Verify Tensorflow
verify()
setup()