This example demonstrates OpenVINO™ toolkit integration with facial detection, using
depth information to approximate distance.
A camera with both depth and RGB sensors is required.
This sample makes use of OpenCV. You can use the OpenCV that is packaged
with OpenVINO by pointing OpenCV_DIR to
A helper namespace
openvino_helpers is used, with a helper class
object_detection encapsulating much of the OpenVINO details:
openvino_helpers::object_detection faceDetector( "face-detection-adas-0001.xml", 0.5 // Probability threshold -- anything with less confidence will be thrown out );
There are two trained model Intermediate Representation files
.bin) that need to be loaded. Pointing to
.xml is enough. These are automatically installed into your build's
object_detectionclass checks that the model includes the required
input/output layers, so feel free to substitute different models.
Each detection has a
confidence score. You can specify how confident you
want the results to be.
Asynchronous detection takes place by queueing a frame and only processing its
results when the next frame is available:
// Wait for the results of the previous frame we enqueued: we're going to process these faceDetector.wait(); auto results = faceDetector.fetch_results(); // Enqueue the current frame so we'd get the results when the next frame comes along! faceDetector.enqueue( image ); faceDetector.submit_request(); // Process the results...
Detected faces are placed into a container and assigned IDs. Some basic effort
is made to keep the creation of new faces to a minimum: previous faces are
compared with new detections to see if the new are simply new positions for the
old. An "intersection over union" (IoU) quotient is calculated and, if over a
threshold, an existing face is moved rather than a new face created.
rect = rect & cv::Rect( 0, 0, image.cols, image.rows ); auto face_ptr = openvino_helpers::find_object( rect, prev_faces ); if( !face_ptr ) // New face face_ptr = std::make_shared< openvino_helpers::detected_object >( id++, rect ); else // Existing face; just update its parameters face_ptr->move( rect );
We make sure to align the depth and color frames, such that the center of each
face's bounding box in the color frame corresponds to the same pixel in the
depth. Then it is easy to simply get the depth:
auto center_x = r.x + r.width / 2; auto center_y = r.y + r.height / 2; auto d = depth_frame.get_distance( center_x, center_y );
Updated over 1 year ago