Add support for Intel OpenVino toolkit 2021.4

hadikoub · hadikoub · commit f5f19ed54273 · 2022-08-16T19:29:30.000+03:00
diff --git a/.gitattributes b/.gitattributes
@@ -1 +1,2 @@
 models/detect_cups/detect_cups.bin filter=lfs diff=lfs merge=lfs -text
+models/saved_model/saved_model.bin filter=lfs diff=lfs merge=lfs -text
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,5 @@
 FROM tiangolo/uvicorn-gunicorn:python3.6 AS uvicorn-builder
-FROM openvino/ubuntu18_runtime:2021.1
+FROM openvino/ubuntu18_runtime:2021.4
 
 USER root
 
diff --git a/README.md b/README.md
@@ -2,7 +2,7 @@
 
 This is a repository for an object detection inference API using the OpenVINO. It's supported on both Windows and Linux Operating systems.
 
-Models in Intermediate Representation(IR) format, converted using the [Intel&reg; OpenVINO&trade; toolkit v2021.1](https://docs.openvinotoolkit.org/2021.1/index.html), can be deployed in this API. Currently, OpenVINO supports conversion for Models trained in several Machine Learning frameworks including Caffe, Tensorflow etc. Please refer to [the OpenVINO documentation](https://docs.openvinotoolkit.org/2021.1/openvino_docs_MO_DG_prepare_model_convert_model_Converting_Model.html) for further details on converting your Model.
+Models in Intermediate Representation(IR) format, converted using [Intel&reg; OpenVINO&trade; toolkit v2021.1](https://docs.openvino.ai/2021.1/index.html) or using [Intel&reg; OpenVINO&trade; toolkit v2021.4](https://docs.openvino.ai/2021.4/index.html), can be deployed in this API. Currently, OpenVINO supports conversion for Models trained in several Machine Learning frameworks including Caffe, Tensorflow etc. Please refer to [the OpenVINO documentation](https://docs.openvinotoolkit.org/2021.1/openvino_docs_MO_DG_prepare_model_convert_model_Converting_Model.html) for further details on converting your Model.
 
 ## Prerequisites
 
diff --git a/models/saved_model/classes.txt b/models/saved_model/classes.txt
@@ -0,0 +1,80 @@
+person
+bicycle
+car
+motorbike
+aeroplane
+bus
+train
+truck
+boat
+traffic light
+fire hydrant
+stop sign
+parking meter
+bench
+bird
+cat
+dog
+horse
+sheep
+cow
+elephant
+bear
+zebra
+giraffe
+backpack
+umbrella
+handbag
+tie
+suitcase
+frisbee
+skis
+snowboard
+sports ball
+kite
+baseball bat
+baseball glove
+skateboard
+surfboard
+tennis racket
+bottle
+wine glass
+cup
+fork
+knife
+spoon
+bowl
+banana
+apple
+sandwich
+orange
+broccoli
+carrot
+hot dog
+pizza
+donut
+cake
+chair
+sofa
+pottedplant
+bed
+diningtable
+toilet
+tvmonitor
+laptop
+mouse
+remote
+keyboard
+cell phone
+microwave
+oven
+toaster
+sink
+refrigerator
+book
+clock
+vase
+scissors
+teddy bear
+hair drier
+toothbrush
diff --git a/models/saved_model/config.json b/models/saved_model/config.json
@@ -0,0 +1,9 @@
+{
+	"predictions": 15,
+	"confidence": 50,
+	"inference_engine_name": "openvino_detection",
+	"framework": "openvino",
+	"type": "detection",
+	"network": "ssd_resnet",
+	"number_of_classes": 2
+}
diff --git a/models/saved_model/saved_model.bin b/models/saved_model/saved_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2bfe8c6ef01eff595bc85397fb75f4677441a3b6c9e11cc6585aaba4f801ebb2
+size 208345728
diff --git a/models/saved_model/saved_model.mapping b/models/saved_model/saved_model.mapping
diff --git a/models/saved_model/saved_model.xml b/models/saved_model/saved_model.xml
diff --git a/src/main/inference/openvino_detection.py b/src/main/inference/openvino_detection.py

Original file line number	Diff line number	Diff line change
`@@ -1 +1,2 @@`
`1`	`1`	`models/detect_cups/detect_cups.bin filter=lfs diff=lfs merge=lfs -text`
	`2`	`+models/saved_model/saved_model.bin filter=lfs diff=lfs merge=lfs -text`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+version https://git-lfs.github.com/spec/v1`
	`2`	`+oid sha256:2bfe8c6ef01eff595bc85397fb75f4677441a3b6c9e11cc6585aaba4f801ebb2`
	`3`	`+size 208345728`