From e8c0d3a76b10c1affccb8f37cefbaf19c81f4501 Mon Sep 17 00:00:00 2001
From: David Chatel <dchatel@live.fr>
Date: Wed, 1 Jan 2025 11:21:14 +0100
Subject: [PATCH] no longer detect faces larger than the input image (they are
 most likely not faces)

---
 utils.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/utils.py b/utils.py
index 742d4b6..543edf9 100644
--- a/utils.py
+++ b/utils.py
@@ -106,7 +106,7 @@ class Face:
             lmk[0,[43,44,46,47],:2].mean(axis=0),
             lmk[0,[30,48,54],:2]
         ]) * 4 * s
-
+        
         self.T2 = np.array([[1, 0, -a], [0, 1, -b], [0, 0, 1]])
         rot = cv2.getRotationMatrix2D((128*s,128*s), 90*i, 1)
         self.R = np.vstack((rot, np.array((0,0,1))))
@@ -135,8 +135,10 @@ def detect_faces(img, threshold):
         
         a,b,c,d = [int(x) for x in (cx - r, cy - r, cx + r, cy + r)]        
         face = Face(img, a, b, c, d)
-
-        faces.append(face)
+        
+        M = estimate_norm(face.kps, 512)
+        if abs(M[0,0]) > 1 and abs(M[1,1]) > 1:
+            faces.append(face)
     return faces
 
 def get_face_mesh(crop: torch.Tensor):