add opencl option for resnet_ssd_face sample

Signed-off-by: Li Peng <peng.li@intel.com>
pull/10492/head
Li Peng 7 years ago
parent 34bfd7ef51
commit 1073175c77
  1. 6
      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())
{

Loading…
Cancel
Save