Fix import pass

PINTO0309 · Apr 26, 2019 · 677b5a9 · 677b5a9
1 parent 72ee91e
commit 677b5a9
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 4 deletions.
diff --git a/openvino_tiny-yolov3_MultiStick_test.py b/openvino_tiny-yolov3_MultiStick_test.py
@@ -1,6 +1,9 @@
 import sys, os, cv2, time, heapq, argparse
 import numpy as np, math
-from openvino.inference_engine import IENetwork, IEPlugin
+try:
+    from armv7l.openvino.inference_engine import IENetwork, IEPlugin
+except:
+    from openvino.inference_engine import IENetwork, IEPlugin
 import multiprocessing as mp
 from time import sleep
 import threading

diff --git a/openvino_tiny-yolov3_test.py b/openvino_tiny-yolov3_test.py
@@ -1,7 +1,10 @@
 import sys, os, cv2, time
 import numpy as np, math
 from argparse import ArgumentParser
-from openvino.inference_engine import IENetwork, IEPlugin
+try:
+    from armv7l.openvino.inference_engine import IENetwork, IEPlugin
+except:
+    from openvino.inference_engine import IENetwork, IEPlugin
 
 m_input_size = 416
 

diff --git a/openvino_yolov3_MultiStick_test.py b/openvino_yolov3_MultiStick_test.py
@@ -1,6 +1,9 @@
 import sys, os, cv2, time, heapq, argparse
 import numpy as np, math
-from openvino.inference_engine import IENetwork, IEPlugin
+try:
+    from armv7l.openvino.inference_engine import IENetwork, IEPlugin
+except:
+    from openvino.inference_engine import IENetwork, IEPlugin
 import multiprocessing as mp
 from time import sleep
 import threading

diff --git a/openvino_yolov3_test.py b/openvino_yolov3_test.py
@@ -1,7 +1,10 @@
 import sys, os, cv2, time
 import numpy as np, math
 from argparse import ArgumentParser
-from openvino.inference_engine import IENetwork, IEPlugin
+try:
+    from armv7l.openvino.inference_engine import IENetwork, IEPlugin
+except:
+    from openvino.inference_engine import IENetwork, IEPlugin
 
 m_input_size = 416