README.md

Engine

Install dependencies

pip install virtualenv
virtualenv venv --python=python3.6
venv/bin/pip install -r requirements.txt

Also install requirements.cpu.txt or requirements.gpu.txt depending on your hardware setup.

Download model and labels

./downloads.sh

More coco models available for download.

Genetate protos

venv/bin/python -m grpc_tools.protoc -I protos --python_out=src --grpc_python_out=src protos/grpc.proto

Run servers

venv/bin/python src/main.py

Run test

venv/bin/python src/test_grpc.py src/test.jpg
venv/bin/python src/test_grpc_webcam.py

Package

venv/bin/pyinstaller src/main.py --onefile --add-data 'src/models/mscoco_label_map.pbtxt:models' --add-data 'src/models/ssd_inception_v2_coco_2017_11_17/frozen_inference_graph.pb:models/ssd_inception_v2_coco_2017_11_17'