diff --git a/samples/dnn/resnet_ssd_face.cpp b/samples/dnn/resnet_ssd_face.cpp index 7a30768ba7..5fe8b8def1 100644 --- 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("opencl")) + { + net.setPreferableTarget(DNN_TARGET_OPENCL); + } + VideoCapture cap; if (parser.get("video").empty()) {