opencv: facedetect: add right/left ear detection

1 files changed, 74 insertions, 4 deletions

diff --git a/ext/opencv/gstfacedetect.cpp b/ext/opencv/gstfacedetect.cpp
index c7cb4d538..1b0d2d474 100644
--- a/ext/opencv/gstfacedetect.cpp
+++ b/ext/opencv/gstfacedetect.cpp
@@ -94,6 +94,8 @@ GST_DEBUG_CATEGORY_STATIC (gst_face_detect_debug);
 #define DEFAULT_NOSE_PROFILE "haarcascade_mcs_nose.xml"
 #define DEFAULT_MOUTH_PROFILE "haarcascade_mcs_mouth.xml"
 #define DEFAULT_EYES_PROFILE "haarcascade_mcs_eyepair_small.xml"
+#define DEFAULT_EAR_L_PROFILE "haarcascade_mcs_leftear.xml"
+#define DEFAULT_EAR_R_PROFILE "haarcascade_mcs_rightear.xml"
 #define DEFAULT_SCALE_FACTOR 1.25
 #if (CV_MAJOR_VERSION >= 4)
 #define DEFAULT_FLAGS CASCADE_DO_CANNY_PRUNING
@@ -254,6 +256,8 @@ gst_face_detect_finalize (GObject * obj)
   g_free (filter->nose_profile);
   g_free (filter->mouth_profile);
   g_free (filter->eyes_profile);
+  g_free (filter->ear_l_profile);
+  g_free (filter->ear_r_profile);
 
   if (filter->cvFaceDetect)
     delete (filter->cvFaceDetect);
@@ -263,6 +267,10 @@ gst_face_detect_finalize (GObject * obj)
     delete (filter->cvMouthDetect);
   if (filter->cvEyesDetect)
     delete (filter->cvEyesDetect);
+  if (filter->cvEarLDetect)
+    delete (filter->cvEarLDetect);
+  if (filter->cvEarRDetect)
+    delete (filter->cvEarRDetect);
 
   G_OBJECT_CLASS (gst_face_detect_parent_class)->finalize (obj);
 }
@@ -374,6 +382,8 @@ gst_face_detect_init (GstFaceDetect * filter)
   filter->nose_profile = g_strconcat (haar_path, DEFAULT_NOSE_PROFILE, NULL);
   filter->mouth_profile = g_strconcat (haar_path, DEFAULT_MOUTH_PROFILE, NULL);
   filter->eyes_profile = g_strconcat (haar_path, DEFAULT_EYES_PROFILE, NULL);
+  filter->ear_r_profile = g_strconcat (haar_path, DEFAULT_EAR_R_PROFILE, NULL);
+  filter->ear_l_profile = g_strconcat (haar_path, DEFAULT_EAR_L_PROFILE, NULL);
   filter->display = TRUE;
   filter->face_detected = FALSE;
   filter->scale_factor = DEFAULT_SCALE_FACTOR;
@@ -390,6 +400,10 @@ gst_face_detect_init (GstFaceDetect * filter)
       gst_face_detect_load_profile (filter, filter->mouth_profile);
   filter->cvEyesDetect =
       gst_face_detect_load_profile (filter, filter->eyes_profile);
+  filter->cvEarLDetect =
+      gst_face_detect_load_profile (filter, filter->ear_l_profile);
+  filter->cvEarRDetect =
+      gst_face_detect_load_profile (filter, filter->ear_r_profile);
 
   gst_opencv_video_filter_set_in_place (GST_OPENCV_VIDEO_FILTER_CAST (filter),
       TRUE);
@@ -593,6 +607,8 @@ gst_face_detect_transform_ip (GstOpencvVideoFilter * base, GstBuffer * buf,
     vector < Rect > mouth;
     vector < Rect > nose;
     vector < Rect > eyes;
+    vector < Rect > r_ears;
+    vector < Rect > l_ears;
     gboolean post_msg = FALSE;
 
     cvtColor (img, filter->cvGray, COLOR_RGB2GRAY);
@@ -646,8 +662,10 @@ gst_face_detect_transform_ip (GstOpencvVideoFilter * base, GstBuffer * buf,
       guint rnx = 0, rny = 0, rnw, rnh;
       guint rmx = 0, rmy = 0, rmw, rmh;
       guint rex = 0, rey = 0, rew, reh;
+      guint relx = 0, rely = 0, relw, relh;
+      guint rerx = 0, rery = 0, rerw, rerh;
       guint rhh = r.height / 2;
-      gboolean have_nose, have_mouth, have_eyes;
+      gboolean have_nose, have_mouth, have_eyes, have_l_ears, have_r_ears;
 
       /* detect face features */
 
@@ -687,11 +705,35 @@ gst_face_detect_transform_ip (GstOpencvVideoFilter * base, GstBuffer * buf,
         have_eyes = FALSE;
       }
 
+      if (filter->cvEarLDetect) {
+        relx = r.x;
+        rely = r.y;
+        relw = r.width;
+        relh = rhh;
+        gst_face_detect_run_detector (filter, filter->cvEarLDetect, mw, mh,
+            Rect (relx, rely, relw, relh), l_ears);
+        have_l_ears = !l_ears.empty ();
+      } else {
+        have_l_ears = FALSE;
+      }
+
+      if (filter->cvEarRDetect) {
+        rerx = r.x;
+        rery = r.y;
+        rerw = r.width;
+        rerh = rhh;
+        gst_face_detect_run_detector (filter, filter->cvEarRDetect, mw, mh,
+            Rect (rerx, rery, rerw, rerh), r_ears);
+        have_r_ears = !r_ears.empty ();
+      } else {
+        have_r_ears = FALSE;
+      }
+
       GST_LOG_OBJECT (filter,
           "%2d/%2" G_GSIZE_FORMAT
-          ": x,y = %4u,%4u: w.h = %4u,%4u : features(e,n,m) = %d,%d,%d", i,
-          faces.size (), r.x, r.y, r.width, r.height, have_eyes, have_nose,
-          have_mouth);
+          ": x,y = %4u,%4u: w.h = %4u,%4u : features(e,n,m) = %d,%d,%d,%d,%d",
+          i, faces.size (), r.x, r.y, r.width, r.height, have_eyes, have_nose,
+          have_mouth, have_r_ears, have_l_ears);
       if (post_msg) {
         s = gst_structure_new ("face",
             "x", G_TYPE_UINT, r.x,
@@ -704,6 +746,10 @@ gst_face_detect_transform_ip (GstOpencvVideoFilter * base, GstBuffer * buf,
           structure_and_message (mouth, "mouth", rmx, rmy, filter, s);
         if (have_eyes)
           structure_and_message (eyes, "eyes", rex, rey, filter, s);
+        if (have_l_ears)
+          structure_and_message (l_ears, "left ears", relx, rely, filter, s);
+        if (have_r_ears)
+          structure_and_message (r_ears, "right ears", rerx, rery, filter, s);
 
         g_value_init (&facedata, GST_TYPE_STRUCTURE);
         g_value_take_boxed (&facedata, s);
@@ -761,6 +807,30 @@ gst_face_detect_transform_ip (GstOpencvVideoFilter * base, GstBuffer * buf,
           axes.height = h;
           ellipse (img, center, axes, 0, 0, 360, Scalar (cr, cg, cb), 1, 8, 0);
         }
+        if (have_r_ears) {
+          Rect sr = r_ears[0];
+
+          w = sr.width / 2;
+          h = sr.height / 2;
+          center.x = cvRound ((rerx + sr.x + w));
+          center.y = cvRound ((rery + sr.y + h));
+          axes.width = w * 1.25; /* tweak for ears form */
+          axes.height = h;
+          ellipse (mtxOrg, center, axes, 0, 0, 360, Scalar (cr, cg, cb), 1, 8,
+              0);
+        }
+        if (have_l_ears) {
+          Rect sr = l_ears[0];
+
+          w = sr.width / 2;
+          h = sr.height / 2;
+          center.x = cvRound ((relx + sr.x + w));
+          center.y = cvRound ((rely + sr.y + h));
+          axes.width = w * 1.25; /* tweak for ears form */
+          axes.height = h;
+          ellipse (mtxOrg, center, axes, 0, 0, 360, Scalar (cr, cg, cb), 1, 8,
+              0);
+        }
       }
       gst_buffer_add_video_region_of_interest_meta (buf, "face",
           (guint) r.x, (guint) r.y, (guint) r.width, (guint) r.height);