add opencl option for resnet_ssd_face sample
authorLi Peng <peng.li@intel.com>
Wed, 3 Jan 2018 15:54:11 +0000 (23:54 +0800)
committerLi Peng <peng.li@intel.com>
Thu, 4 Jan 2018 10:40:56 +0000 (18:40 +0800)
Signed-off-by: Li Peng <peng.li@intel.com>
samples/dnn/resnet_ssd_face.cpp

index 7a30768..5fe8b8d 100644 (file)
@@ -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())
     {