Merge pull request #279 from biometrics/expose_pp5_context

Expose certain PP5 context properties in PP5EnrollTransform

Merge pull request #279 from biometrics/expose_pp5_context
Expose certain PP5 context properties in PP5EnrollTransform
Josh Klontz
2 parents c822d46f 28903cdf
Showing 1 changed file with 63 additions and 13 deletions
openbr/plugins/pp5.cpp
@@ -46,7 +46,7 @@ class PP5Initializer : public Initializer
     {
         TRY(ppr_initialize_sdk(qPrintable(Globals->sdkPath + "/share/openbr/models/pp5/"), my_license_id, my_license_key))
         Globals->abbreviations.insert("PP5","Open+Expand+PP5Enroll!PP5Gallery");
-        Globals->abbreviations.insert("PP5Register", "Open+PP5Enroll(true,requireLandmarks=true)+RenameFirst([eyeL,PP5_Landmark0_Right_Eye],Affine_0)+RenameFirst([eyeR,PP5_Landmark1_Left_Eye],Affine_1)");
+        Globals->abbreviations.insert("PP5Register", "Open+PP5Enroll(true,true,0.02,5,Frontal,1)+RenameFirst([eyeL,PP5_Landmark0_Right_Eye],Affine_0)+RenameFirst([eyeR,PP5_Landmark1_Left_Eye],Affine_1)");
         Globals->abbreviations.insert("PP5CropFace", "Open+PP5Enroll(true)+RenameFirst([eyeL,PP5_Landmark0_Right_Eye],Affine_0)+RenameFirst([eyeR,PP5_Landmark1_Left_Eye],Affine_1)+Affine(128,128,0.25,0.35)+Cvt(Gray)");
     }
@@ -67,24 +67,25 @@ struct PP5Context
 {
     ppr_context_type context;
-    PP5Context()
+    PP5Context(bool detectOnly = false, float adaptiveMinSize = 0.01f, int minSize = 4, ppr_landmark_range_type landmarkRange = PPR_LANDMARK_RANGE_COMPREHENSIVE, int searchPruningAggressiveness = 0)
     {
         ppr_settings_type default_settings = ppr_get_default_settings();
         default_settings.detection.adaptive_max_size = 1.f;
-        default_settings.detection.adaptive_min_size = 0.01f;
+        default_settings.detection.adaptive_min_size = adaptiveMinSize;
         default_settings.detection.detect_best_face_only = !Globals->enrollAll;
         default_settings.detection.enable = 1;
-        default_settings.detection.min_size = 4;
+        default_settings.detection.min_size = minSize;
+        default_settings.detection.search_pruning_aggressiveness = searchPruningAggressiveness;
         default_settings.detection.use_serial_face_detection = 1;
         default_settings.landmarks.enable = 1;
-        default_settings.landmarks.landmark_range = PPR_LANDMARK_RANGE_COMPREHENSIVE;
+        default_settings.landmarks.landmark_range = landmarkRange;
         default_settings.landmarks.manually_detect_landmarks = 0;
-        default_settings.recognition.automatically_extract_templates = 1;
-        default_settings.recognition.enable_comparison = 1;
-        default_settings.recognition.enable_extraction = 1;
+        default_settings.recognition.automatically_extract_templates = !detectOnly;
+        default_settings.recognition.enable_comparison = !detectOnly;
+        default_settings.recognition.enable_extraction = !detectOnly;
         default_settings.recognition.num_comparison_threads = 1;
         default_settings.recognition.recognizer = PPR_RECOGNIZER_MULTI_POSE;
         TRY(ppr_initialize_context(default_settings, &context))
@@ -249,6 +250,8 @@ struct PP5Context
 /*!
  * \ingroup transforms
  * \brief Enroll faces in PP5
+ *
+ * See PittPatt documentation for the relationship between minSize and pixel IPD.
  * \author Josh Klontz \cite jklontz
  * \author E. Taborsky \cite mmtaborsky
  */
@@ -256,11 +259,55 @@ class PP5EnrollTransform : public UntrainableMetaTransform
 {
     Q_OBJECT
     Q_PROPERTY(bool detectOnly READ get_detectOnly WRITE set_detectOnly RESET reset_detectOnly STORED false)
-    BR_PROPERTY(bool, detectOnly, false)
     Q_PROPERTY(bool requireLandmarks READ get_requireLandmarks WRITE set_requireLandmarks RESET reset_requireLandmarks STORED false)
+    Q_PROPERTY(float adaptiveMinSize READ get_adaptiveMinSize WRITE set_adaptiveMinSize RESET reset_adaptiveMinSize STORED false)
+    Q_PROPERTY(int minSize READ get_minSize WRITE set_minSize RESET reset_minSize STORED false)
+    Q_PROPERTY(LandmarkRange landmarkRange READ get_landmarkRange WRITE set_landmarkRange RESET reset_landmarkRange STORED false)
+    Q_PROPERTY(int searchPruningAggressiveness READ get_searchPruningAggressiveness WRITE set_searchPruningAggressiveness RESET reset_searchPruningAggressiveness STORED false)
+
+public:
+    enum LandmarkRange
+    {
+        Frontal       = PPR_LANDMARK_RANGE_FRONTAL,
+        Extended      = PPR_LANDMARK_RANGE_EXTENDED,
+        Full          = PPR_LANDMARK_RANGE_FULL,
+        Comprehensive = PPR_LANDMARK_RANGE_COMPREHENSIVE
+    };
+    Q_ENUMS(LandmarkRange)
+
+private:
+    BR_PROPERTY(bool, detectOnly, false)
     BR_PROPERTY(bool, requireLandmarks, false)
+    BR_PROPERTY(float, adaptiveMinSize, 0.01f)
+    BR_PROPERTY(int, minSize, 4)
+    BR_PROPERTY(LandmarkRange, landmarkRange, Comprehensive)
+    BR_PROPERTY(int, searchPruningAggressiveness, 0)
+
     Resource<PP5Context> contexts;
+    struct PP5ContextMaker : public ResourceMaker<PP5Context>
+    {
+        PP5ContextMaker(PP5EnrollTransform *pp5EnrollTransform)
+            : pp5EnrollTransform(pp5EnrollTransform) {}
+
+    private:
+        PP5EnrollTransform *pp5EnrollTransform;
+
+        PP5Context *make() const
+        {
+            return new PP5Context(pp5EnrollTransform->detectOnly,
+                                  pp5EnrollTransform->adaptiveMinSize,
+                                  pp5EnrollTransform->minSize,
+                                  (ppr_landmark_range_type) pp5EnrollTransform->landmarkRange,
+                                  pp5EnrollTransform->searchPruningAggressiveness);
+        }
+    };
+
+    void init()
+    {
+        contexts.setResourceMaker(new PP5ContextMaker(this));
+    }
+
     void project(const Template &src, Template &dst) const
     {
         if (Globals->enrollAll)
@@ -293,10 +340,13 @@ class PP5EnrollTransform : public UntrainableMetaTransform
                 for (int i=0; i<face_list.length; i++) {
                     ppr_face_type face = face_list.faces[i];
-                    int extractable;
-                    TRY(ppr_is_template_extractable(context->context, face, &extractable))
-                    if (!extractable && !detectOnly) continue;
-                    else foundFace = true;
+                    if (!detectOnly) {
+                        int extractable;
+                        TRY(ppr_is_template_extractable(context->context, face, &extractable))
+                        if (!extractable)
+                            continue;
+                    }
+                    foundFace = true;
                     cv::Mat m;
                     if (detectOnly) {