To use on devices w/o opencl support.

Author: solderzzc

src/embedding/FaceProcessing.py

@@ -22,6 +22,7 @@
 from scipy.misc import imread, imresize
 import sklearn.preprocessing
 
+global mx
 global globGraph
 globGraph = None
 global mod
@@ -32,6 +33,7 @@
 DEBUG = False
 
 DATA_RUNTIME_FOLDER = os.getenv('DATA_RUNTIME_FOLDER', '/data/runtime')
+HAS_OPENCL = os.getenv('HAS_OPENCL', 'true')
 
 def load_graph(frozen_graph_filename):
     return None
@@ -108,6 +110,25 @@ def get_model(ctx, image_size, model_str, layer):
 def init_embedding_processor():
     global mod2
     global mod3
+
+    if HAS_OPENCL == 'false':
+        global mx
+        import mxnet as mx
+        print('need init mxnet')
+
+        mod2 = None
+        if os.path.isfile(DATA_RUNTIME_FOLDER+'/model-0000.params'):
+            ctx = mx.cpu(0)
+            mod3 = get_model(ctx, [112,112], DATA_RUNTIME_FOLDER+'/model,0', 'fc1')
+            print('backup model loaded')
+            return mod3
+
+        mod = FaceProcessing.init_embedding_processor()
+        print("start to warm up")
+        embedding = featureCalculation2(os.path.join(BASEDIR,"image","Mike_Alden_0001_tmp.png"))
+        print("warmed up")
+        return mod3
+
     if os.path.isfile(DATA_RUNTIME_FOLDER+'/net2'):
         global __t
         global graph_runtime

src/embedding/upload_api-v2.py

@@ -130,6 +130,7 @@
 # run as worker only
 CLUSTER_WORKERONLY = os.getenv('CLUSTER_WORKERONLY', False)
 
+HAS_OPENCL = os.getenv('HAS_OPENCL', 'true')
 SAVE_ORIGINAL_FACE = False
 original_face_img_path = os.path.join(BASEDIR, 'data', 'original_face_img')
 if not os.path.exists(original_face_img_path):
@@ -142,6 +143,14 @@
 
 counter = 0
 
+if HAS_OPENCL == 'false':
+    print('need init mxnet')
+    mod = FaceProcessing.init_embedding_processor()
+    print("start to warm up")
+    embedding = FaceProcessing.FaceProcessingImageData2(os.path.join(BASEDIR,"image","Mike_Alden_0001_tmp.png"))
+    print("warmed up")
+
+
 def featureCalculation(imgpath):
     img = misc.imread(os.path.expanduser(imgpath))
     prewhitened = facenet.prewhiten(img)
@@ -1638,11 +1647,11 @@ def setup(sender=None, **kwargs):
         check_groupid_changed()
         init_fs()
 
-
-        mod = FaceProcessing.init_embedding_processor()
-        print("start to warm up")
-        embedding = featureCalculation2(os.path.join(BASEDIR,"image","Mike_Alden_0001_tmp.png"))
-        print("warmed up")
+        if HAS_OPENCL == 'true':
+            mod = FaceProcessing.init_embedding_processor()
+            print("start to warm up")
+            embedding = featureCalculation2(os.path.join(BASEDIR,"image","Mike_Alden_0001_tmp.png"))
+            print("warmed up")
         #if embedding is not None:
         #    print("worker embedding ready")
 

start_service.sh

@@ -1,6 +1,15 @@
 #!/bin/bash
 source ./env/common.sh
 
+if [ -f /system/vendor/lib64/libOpenCL.so ]
+then
+  export HAS_OPENCL=1
+  echo 'has opencl supporting'
+else
+  export HAS_OPENCL=0
+  echo 'no opencl supporting'
+fi
+
 redis-server --maxmemory 40mb --maxmemory-policy allkeys-lru --save "" --appendonly no --dbfilename "" &
 mosquitto &