jiekexian / objectron-3d-object-detection-openvino

Sample implementation of 3D object detection with Intel OpenVINO

Overview

This is a sample implementaion of 3D object detection application using Intel(r) Distribution of OpenVINO(tm) Toolkit. The program uses objectron model from Google mediapipe project to detect the objects from the input images.
In this project, I took the objectron model in TF SavedModel format from PINTO model zoo in the GitHub.

これはIntel(r) Distribution of OpenVINO(tm) Toolkitを使用した3D物体検出アプリケーションの参考実装です。プログラムはGoogle mediapipeプロジェクトのobjectronモデルを使用しています。
このプロジェクトではPINTOさんのPINTO model zooからTF SavedModelフォーマットのモデルデータを使用させて頂いています。

Example of Detection result

1. Prerequisites

• Intel Distribution of OpenVINO toolkit 2020.2 (or newer)

2. Download Objectron TensorFlow model from PINTO model zoo

Linux compatible environment is required to run download.sh script to get the model

2.1 Download the model download script

Download download.sh from https://github.com/PINTO0309/PINTO_model_zoo/tree/master/36_Objectron/01_float32

2.2 Run download.sh to get saved_model_object_detection_3d_chair/*

 ./download.sh

You'll get following files (and the other model files too).

saved_model_object_detection_3d_chair/
+ variables/
| + variables.data-00000-of-00001
| + variables.index
+ saved_model.pb

3. Convert objectron TF model into OpenVINO IR model

Use Model Optimizer (MO) to convert the TF model into IR model.

Linux

python3 ${INTEL_OPENVINO_DIR}/deployment_tools/model_optimizer/mo.py \
 --saved_model_dir saved_model_object_detection_3d_chair \
 --scale 255 \
 --output_dir objectron_chair \
 --data_type FP16

Windows

python "%INTEL_OPENVINO_DIR%\deployment_tools\model_optimizer\mo.py" ^
 --saved_model_dir saved_model_object_detection_3d_chair ^
 --scale 255 ^
 --output_dir objectron_chair ^
 --data_type FP16

An IR model ./objectron_chair/saved_model.xml|.bin will be generated.

4. Run sample program

(Linux) python3 objectron.py [-m model.xml] [-i input_image] [--heatmap]
(Win10) python objectron.py [-m model.xml] [-i input_image] [--heatmap]

Options
-m: IR model path (default: ./objectron_chair/saved_model.xml)
-i: Input image path. A webCam will be used as the input when you specify cam (default: chair.jpg)
--heatmap: Display heatmap

Command line example:

 $ python3 objectron.py -m ./objectron_chair/object_detection_3d_chair.xml -i cam --heatmap

5. Test Environment

• Ubuntu 18.04 / Windows 10 1909
• OpenVINO 2020.3 LTS

See Also

• Google AI Blog, Real-Time 3D Object Detection on Mobile Devices with MediaPipe
• GitHub, Google mediapipe project
• GitHub, PINTO model zoo
• Intel(r) Distribution of OpenVINO(tm) Toolkit