Object Detection API
49 |Upload a color photo file.
50 |
51 |
55 |
56 |
57 | 58 | 59 |
60 | {% if result|length > 0 %}
61 | 
63 |
81 |
84 |
85 |
86 |
87 |
--------------------------------------------------------------------------------
/setup/template-setenv-local.sh:
--------------------------------------------------------------------------------
1 | #!/bin/bash
2 |
3 | #
4 | # Copyright 2018 Google LLC
5 | #
6 | # Licensed under the Apache License, Version 2.0 (the "License");
7 | # you may not use this file except in compliance with the License.
8 | # You may obtain a copy of the License at
9 | #
10 | # https://www.apache.org/licenses/LICENSE-2.0
11 | #
12 | # Unless required by applicable law or agreed to in writing, software
13 | # distributed under the License is distributed on an "AS IS" BASIS,
14 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | # See the License for the specific language governing permissions and
16 | # limitations under the License.
17 | #
18 |
19 | ##################################################################################
20 | # Environment settings that are unique to a development machine - these
21 | # do not get committed into repo and allow multi-user development. Example of this
22 | # can be found in "src/setup" folder.
23 | ##################################################################################
24 |
25 | set -u # This prevents running the script if any of the variables have not been set
26 | set -e # Exit if error is detected during pipeline execution
27 |
28 | echo "setenv-local.sh: start..."
29 |
30 | ### Automatically generate unique project ID for the first run and save it into a file. Later read it from file
31 | PROJECT_NAME_FILE="$PROJECT_DIR/project-id.sh"
32 |
33 | if [ -f "$PROJECT_NAME_FILE" ] ; then
34 | echo "Sourcing existing project file '$PROJECT_NAME_FILE'..."
35 | source $PROJECT_NAME_FILE
36 | else
37 | # Infer current project ID from the environment
38 | export PROJECT=$(gcloud info | grep "Project:" | sed -n -e "s/Project: \[//p" | sed -n -e "s/\]//p")
39 | fi
40 |
41 | echo "PROJECT='$PROJECT'"
42 | gcloud config set project "$PROJECT"
43 |
44 | ### Serial number of the car to distinguish it from all other cars possibly on the same project
45 | export CAR_ID=2
46 |
47 | ### How many training steps to take during TensorFlow model training
48 | TRAINING_STEPS=8000
49 |
50 | ### This controls which inference VM REST API will be used by the Driving Controller
51 | # true - use existing inference VM from the DEMO project
52 | # false - use inference VM within the same project as Driving Controller
53 | USE_DEMO_INFERENCE="false"
54 |
55 | ### Where do we want Driving Controller to be deployed? (current VM or App Engine)
56 | # true - deploy in local VM
57 | # false - deploy in App Engine
58 | DRIVING_CONTROLLER_LOCAL="true"
59 |
60 | ### This controls certain automated tasks and allows the script to create resources on behalf of the user
61 | FOUR_HOURS_HACKATHON="false"
62 |
63 | if $FOUR_HOURS_HACKATHON ; then
64 | AUTO_CREATE_IP="true"
65 | AUTO_CREATE_FIREWALL="true"
66 | SKIP_MANUAL_IMAGE_ANNOTATION="true"
67 | fi
68 |
69 | ### Used for multiple ML models to be deployed and compared against each other (this is added to VM names, IP names, GCS bucket names, etc.)
70 | export VERSION=50
71 |
72 | ### These are Region and Zone where you want to run your car controller - feel free to change as you see fit
73 | export REGION="us-central1"
74 | export ZONE="us-central1-f"
75 | export REGION_LEGACY="us-central" # there are corner cases where gcloud still references the legacy nomenclature
76 |
77 | echo "setenv-local.sh: done"
--------------------------------------------------------------------------------
/cloud/ml/inference/run.sh:
--------------------------------------------------------------------------------
1 | #!/bin/bash
2 |
3 | ##################################################
4 | # Run Object Detection based on the model trained earlier by "transferred-learning"
5 | # Based on https://github.com/GoogleCloudPlatform/tensorflow-object-detection-example
6 | #
7 | # This code will run on a special GCE VM $ML_VM
8 | ##################################################
9 |
10 | #
11 | # Copyright 2018 Google LLC
12 | #
13 | # Licensed under the Apache License, Version 2.0 (the "License");
14 | # you may not use this file except in compliance with the License.
15 | # You may obtain a copy of the License at
16 | #
17 | # https://www.apache.org/licenses/LICENSE-2.0
18 | #
19 | # Unless required by applicable law or agreed to in writing, software
20 | # distributed under the License is distributed on an "AS IS" BASIS,
21 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
22 | # See the License for the specific language governing permissions and
23 | # limitations under the License.
24 | #
25 |
26 | set -u # This prevents running the script if any of the variables have not been set
27 | set -e # Exit if error is detected during pipeline execution
28 |
29 | source ../../../setenv-global.sh
30 | source ../setenv-ml.sh
31 |
32 | export PATH_TO_LABELS=$MODEL_BASE/object_detection/data/$LABEL_MAP
33 | export PATH_TO_CKPT=$PROJECT_DIR/checkpoint_graph_def
34 |
35 | ##################################################
36 | # This changes TF inference model to be used for web app
37 | ##################################################
38 | set_inference_model() {
39 | echo_my "Changing inference model..."
40 |
41 | # Point to the trained model
42 | FROM=$CWD/tmp/import
43 | mkdir -p $FROM
44 |
45 | echo_my "Download label map..."
46 | gsutil cp $LABEL_MAP_GCS $FROM/$LABEL_MAP
47 | PATH_TO_LABELS=${MODEL_BASE}/object_detection/data/$LABEL_MAP
48 | sudo ln -sf $FROM/$LABEL_MAP $PATH_TO_LABELS
49 | echo_my "Label map is setup at '$PATH_TO_LABELS'"
50 |
51 | echo_my "Download frozen inference graph from GCS '$FROZEN_INFERENCE_GRAPH_GCS'..."
52 | gsutil cp $FROZEN_INFERENCE_GRAPH_GCS $FROM/frozen_inference_graph.pb
53 |
54 | local DESTINATION_GRAPH=$PATH_TO_CKPT/frozen_inference_graph.pb
55 | if ! [ -d $PATH_TO_CKPT ] ; then
56 | mkdir $PATH_TO_CKPT
57 | fi
58 | sudo ln -sf $FROM/frozen_inference_graph.pb $DESTINATION_GRAPH
59 | echo_my "Frozen inference graph is setup at '$DESTINATION_GRAPH'"
60 | }
61 |
62 | ###############################################
63 | # MAIN
64 | ###############################################
65 | print_header "Starting Object Detection App..."
66 |
67 | mkdir -p tmp
68 | CWD=$(pwd)
69 | INSTALL_FLAG=$CWD/tmp/install.marker
70 |
71 | if [ -f "$INSTALL_FLAG" ]; then
72 | echo_my "File '$INSTALL_FLAG' was found = > no need to do the install since it already has been done."
73 | else
74 | set_inference_model
75 | echo_my "Setting up web app..."
76 | sudo pip install --upgrade Flask==0.12.2 WTForms==2.1 Flask_WTF==0.14.2 Werkzeug==0.12.2 numpy google-cloud-storage
77 | touch $INSTALL_FLAG
78 | fi
79 |
80 | set_python_path
81 | cd $CWD/python
82 |
83 | if [ -f "*.jpg" ] ; then
84 | # Remove old jpg files
85 | rm *.jpg
86 | fi
87 |
88 | if [ -f "nohup.out" ] ; then
89 | rm -rf nohup.out
90 | fi
91 |
92 | echo_my "Running webapp: Change USERNAME and PASSWORD in decorator.py..."
93 | # -u disables line buffering in python and shows everything in the nohup.out
94 | nohup python -u ./app.py &
95 |
96 | tail -f nohup.out
97 |
98 | echo_my "To watch output of the app log, use command: original
62 | 64 | original 65 | 66 | {% for name, img in result.iteritems() %} 67 | {% if name != 'original' and name != 'response_msg' %} 68 | {{ name }} 70 | 71 | {% endif %} 72 | {% endfor %} 73 |
74 |
75 |
79 | {% endif %}
80 | 76 | {{ result['response_msg'] }} 77 |
78 |