Night-time Object Detection: Tensorflow API

Disclaimer: This is not an official Google product.

This is an example application demonstrating how TensorFlow Object Detection API and pretrained models can be used to create a general object detection service.

The base code is forked from https://github.com/wingedrasengan927/Tensorflow-2-Object-Detection-API-Flask-Application

Products

• TensorFlow
• Google Compute Engine

Language

• Python

Prerequisites

1. A Google Cloud Platform Account
2. A new Google Cloud Platform Project for this lab with billing enabled

Do this first

First you launch a GCE instance with the following configuration.

• vCPU x 8
• Memory 8GB
• Debian GNU/Linux 9 as a guest OS
• Allow HTTP traffic
• Assign a static IP address

You can leave other settings as default. Once the instance has started, log in to the guest OS using SSH and change the OS user to root.

$ sudo -s

All remaining operations should be done from the root user.

Install packages

# apt-get update
# apt-get install -y protobuf-compiler python3-pil python3-lxml python3-pip python3-dev git
# pip3 install -U pip
# python3 -m pip install Flask==1.1.1 WTForms==2.2.1 Flask_WTF==0.14.2 Werkzeug==0.16.0 tensorflow==2.0.0

Install the Object Detection API library

# cd /opt
# git clone https://github.com/tensorflow/models
# cd models/research
# protoc object_detection/protos/*.proto --python_out=.

Install the demo application

# cd $HOME
# git clone https://github.com/sohiniroych/tensorflow-object-detection-example
# cp -a tensorflow-object-detection-example/object_detection_app_p1 /opt/
# chmod u+x /opt/object_detection_app_p1/app.py

This application provides a simple user authentication mechanism. You can change the username and password by modifying the following part in /opt/object_detection_app_p1/decorator.py.

USERNAME = 'username'
PASSWORD = 'passw0rd'

Launch the demo application

# python3 /opt/object_detection_app_p1/app.py

You have to wait around 60secs for the application to finish loading the pretrained model graph. You'll see the message Running on http://0.0.0.0:80/ (Press CTRL+C to quit) when it's ready.

Now you can access the instance's static IP address using a web browser. When you upload an image file with a jpeg, jpg, or png extension, the application shows the result of the object detection inference. The inference may take up to 30 seconds, depending on the image.

How to use different models

There are pretrained models that can be used by the application. They have diffrent characteristics in terms of accuracy and speed. You can change the model used by the application with the following steps.

1. Choose one of COCO-trained models from [Tensorflow detection model zoo][7]. (The "Outputs" column should be "Boxes".)
2. Copy an URL of the model from a link on the "Model name" column.
3. Open /opt/object_detection_app_p3/app.py and replace the URL in the following part.

MODEL_URL = 'http://download.tensorflow.org/models/object_detection/faster_rcnn_resnet50_coco_2018_01_28.tar.gz'