-
Notifications
You must be signed in to change notification settings - Fork 5
/
tensorObjectDetection.sh
88 lines (67 loc) · 2.29 KB
/
tensorObjectDetection.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
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
#!/bin/bash
# Copyright (c) 2018 Jegathesan Shanmugam
# Released under the MIT License (MIT)
# https://github.com/nullbyte91/Simple-Sh-DataScience/blob/master/LICENSE.md
# Title : tensorObjectDetection.sh
# Description : This script to install tensorflow object detection on virtual env(https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md)
# author : Jegathesan Shanmugam
python3="python3-dev"
pip3="python3-pip"
virtEnv="virtualenv"
deepSpeech="deepspeech"
protobuf="protobuf-compiler"
pythonPil="python-pil"
pythonLxml="python-lxml"
pythonTk="python-tk"
function virt_activate()
{
echo "$1"
source virtualenv_activate.sh $1
}
#MainStartsHere
source ./helper_function.sh
#systemBasicUpdates
read -r -p "Select the folder to work on objectDetection:[/home/user/objectDetection]" workPath
if [ "${workPath}" == "" ]
then
objectDetection="/home/$USER/objectDetection"
else
objectDetection=${workPath}
fi
if [[ ! -e ${objectDetection} ]]
then
mkdir -p ${objectDetection}
fi
#Before Push to your working dir copy virtualenv_activate.sh
cp virtualenv_activate.sh ${objectDetection}
# We need to create a virtual env for deep Speech
echo "Using Python 3"
#Install Python3 Package
installAptPackages ${python3}
#Install Python3 pip package manager
installAptPackages ${pip3}
#Install virtualenv Pip package
installPipPackages "3" ${virtEnv}
#Create a virtual env
virtTemp="${objectDetection}/virt_temp"
echo "creating a virtual env.......${virtTemp}"
mkdir -p virt_temp
virtualenv --system-site-packages -p python3 ${virtTemp}
#Activate the Virtual Env
virt_activate ${virtTemp}
installAptPackages ${protobuf}
installAptPackages ${pythonPil}
installAptPackages ${pythonLxml}
installAptPackages ${pythonTk}
#Install deepSpeech pip Package
pip3 install -r requirementsFiles/objectDetectionReq.txt
#protobuf-compiler installation and compile
git clone https://github.com/tensorflow/models.git
pushd models/research/
wget -O protobuf.zip https://github.com/google/protobuf/releases/download/v3.0.0/protoc-3.0.0-linux-x86_64.zip
unzip protobuf.zip
./bin/protoc object_detection/protos/*.proto --python_out=.
#Add Libraries to PYTHONPATH
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
# Testing
python object_detection/builders/model_builder_test.py