Add support for Person Head Detection

Author: Myzhar

CHANGELOG.rst

@@ -13,6 +13,7 @@ ZED SDK v3.8
 - Add `od-prediction-timeout-s` parameter [Thx @ryanppeters]
 - Add `od-allow-reduced-precision-inference` parameter [Thx @ryanppeters]
 - Add new field `frame_id` to `GstZedSrcMeta`  in order to track the meta/buffer throughout the GStreamer pipeline (when working with source code) [Thx @ryanppeters]
+- Add support for new Object Detection models in `od-detection-model` (Person Head and Person Head ACCURATE)
 
 ZED SDK v3.7
 -------------

README.md

@@ -163,7 +163,7 @@ Most of the parameters follow the same name as the C++ API. Except that `_` is r
  * `od-enabled`: Enable Object Detection - {TRUE, FALSE}
  * `od-image-sync`: Set to TRUE to enable Object Detection frame synchronization - {TRUE, FALSE}
  * `od-enable-tracking`: Set to TRUE to enable tracking for the detected objects - {TRUE, FALSE}
- * `od-detection-model`: Object Detection Model - {(0): Object Detection Multi class, (1): Object Detection Multi class ACCURATE, (2): Skeleton tracking FAST, (3): Skeleton tracking ACCURATE, (4): Object Detection Multi class MEDIUM, (5): Skeleton tracking MEDIUM}
+ * `od-detection-model`: Object Detection Model - {(0): Object Detection Multi class, (1): Object Detection Multi class ACCURATE, (2): Skeleton tracking FAST, (3): Skeleton tracking ACCURATE, (4): Object Detection Multi class MEDIUM, (5): Skeleton tracking MEDIUM, (6): Person Head, (7): Person Head ACCURATE}
  * `od-confidence`: Minimum Detection Confidence - {0,100}
  * `od-max-range`: Maximum Detection Range - [-1,20000]
  * `od-body-fitting`: Set to TRUE to enable body fitting for skeleton tracking - {TRUE, FALSE}

gst-zed-src/gstzedsrc.cpp

@@ -156,7 +156,9 @@ typedef enum {
     GST_ZEDSRC_OD_HUMAN_BODY_FAST = 2,
     GST_ZEDSRC_OD_HUMAN_BODY_ACCURATE  = 3,
     GST_ZEDSRC_OD_MULTI_CLASS_BOX_MEDIUM = 4,
-    GST_ZEDSRC_OD_HUMAN_BODY_MEDIUM  = 5
+    GST_ZEDSRC_OD_HUMAN_BODY_MEDIUM  = 5,
+    GST_ZEDSRC_OD_PERSON_HEAD_BOX = 6,
+    GST_ZEDSRC_OD_PERSON_HEAD_BOX_ACCURATE = 7,
 } GstZedSrcOdModel;
 
 typedef enum {
@@ -448,6 +450,12 @@ static GType gst_zedsrc_od_model_get_type (void)
             { GST_ZEDSRC_OD_HUMAN_BODY_MEDIUM,
               "Keypoints based, specific to human skeleton, compromise between accuracy and speed",
               "Skeleton tracking MEDIUM"  },
+            { GST_ZEDSRC_OD_PERSON_HEAD_BOX ,
+              "Bounding Box detector specialized in person heads, particulary well suited for crowded environments, the person localization is also improved",
+              "Person Head"  },
+            { GST_ZEDSRC_OD_PERSON_HEAD_BOX_ACCURATE ,
+              "Bounding Box detector specialized in person heads, particulary well suited for crowded environments, the person localization is also improved, more accurate but slower than the base model",
+              "Person Head ACCURATE"  },
             { 0, NULL, NULL },
         };