From 41ee68f56568b2af182b01c11472e375ea0123ff Mon Sep 17 00:00:00 2001 From: Ben Hoff Date: Thu, 27 Jun 2019 02:42:31 -0400 Subject: [PATCH] add in macos handeling for `auto_annotation` (#524) --- cvat/apps/auto_annotation/inference_engine.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/cvat/apps/auto_annotation/inference_engine.py b/cvat/apps/auto_annotation/inference_engine.py index 6961511b..94c977fa 100644 --- a/cvat/apps/auto_annotation/inference_engine.py +++ b/cvat/apps/auto_annotation/inference_engine.py @@ -6,6 +6,7 @@ from openvino.inference_engine import IENetwork, IEPlugin import subprocess import os +import platform _IE_PLUGINS_PATH = os.getenv("IE_PLUGINS_PATH", None) @@ -27,6 +28,8 @@ def make_plugin(): plugin.add_cpu_extension(os.path.join(_IE_PLUGINS_PATH, 'libcpu_extension_avx2.so')) elif (_check_instruction('sse4')): plugin.add_cpu_extension(os.path.join(_IE_PLUGINS_PATH, 'libcpu_extension_sse4.so')) + elif platform.system() == 'Darwin': + plugin.add_cpu_extension(os.path.join(_IE_PLUGINS_PATH, 'libcpu_extension.dylib')) else: raise Exception('Inference engine requires a support of avx2 or sse4.')