Support for OpenVINO. x86 only

PINTO0309 · Feb 15, 2019 · d96a010 · d96a010
1 parent 17cc111
commit d96a010
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/main_openvino.py b/main_openvino.py
@@ -33,8 +33,12 @@ def main(camera_FPS, camera_width, camera_height, inference_scale, threshold, de
 
         # DOC
         print("DOC Model loading...")
-        model_xml="irmodels/tensorflow/FP32/weights.xml"
-        model_bin="irmodels/tensorflow/FP32/weights.bin"
+        if device == "MYRIAD":
+            model_xml="irmodels/tensorflow/FP16/weights.xml"
+            model_bin="irmodels/tensorflow/FP16/weights.bin"
+        else:
+            model_xml="irmodels/tensorflow/FP32/weights.xml"
+            model_bin="irmodels/tensorflow/FP32/weights.bin"
         net = IENetwork(model=model_xml, weights=model_bin)
         plugin = IEPlugin(device=device)
         if device == "CPU":