add opencl option for resnet_ssd_face sample

pli2-intel · pli2-intel · commit 1073175c7788 · 2018-01-04T18:40:56.000+08:00
Signed-off-by: Li Peng &lt;peng.li@intel.com&gt;
diff --git a/samples/dnn/resnet_ssd_face.cpp b/samples/dnn/resnet_ssd_face.cpp
@@ -30,6 +30,7 @@ const char* params
       "{ model          |       | model weights (res10_300x300_ssd_iter_140000.caffemodel) }"
       "{ camera_device  | 0     | camera device number }"
       "{ video          |       | video or image for detection }"
+      "{ opencl         | false | enable OpenCL }"
       "{ min_confidence | 0.5   | min confidence       }";
 
 int main(int argc, char** argv)
@@ -62,6 +63,11 @@ int main(int argc, char** argv)
         exit(-1);
     }
 
+    if (parser.get<bool>("opencl"))
+    {
+        net.setPreferableTarget(DNN_TARGET_OPENCL);
+    }
+
     VideoCapture cap;
     if (parser.get<String>("video").empty())
     {
