Only include faces with the 3 landmarks (if requireLandmarks = true)

Austin Blanton
1 parent 1a5a4da1
Showing 1 changed file with 12 additions and 5 deletions
openbr/plugins/pp5.cpp
@@ -309,14 +309,21 @@ class PP5EnrollTransform : public UntrainableMetaTransform
                     dst.file = src.file;
  
                     dst.file.append(PP5Context::toMetadata(face));
+                    dst += m;
                     if (requireLandmarks) {
-                        QPointF pt = dst.file.get<QPointF>("PP5_Landmark0_Right_Eye");
-                        if (std::isnan(pt.x()) && std::isnan(pt.y())){
-                            dst.file.fte = true;
+                        QPointF right = dst.file.get<QPointF>("PP5_Landmark0_Right_Eye");
+                        QPointF left = dst.file.get<QPointF>("PP5_Landmark1_Left_Eye");
+                        QPointF nose = dst.file.get<QPointF>("PP5_Landmark2_Nose_Base");
+                        if (dst.file.get<int>("PP5_Face_NumLandmarks") >= 3 &&
+                            !( std::isnan(right.x()) && std::isnan(right.y()) &&
+                               std::isnan(left.x()) && std::isnan(left.y()) &&
+                               std::isnan(nose.x()) && std::isnan(nose.y()) ))
+                        {
+                            dstList.append(dst);
                         }
+                    } else {
+                        dstList.append(dst);
                     }
-                    dst += m;
-                    dstList.append(dst);
  
                     // Found a face, nothing else to do (if we aren't trying to find multiple faces).
                     if (!Globals->enrollAll)