Update face_analysis.py

face_analysis.py

@@ -4,6 +4,7 @@ from facenet_pytorch import InceptionResnetV1
 from sklearn.cluster import DBSCAN
 import os
 import shutil
+import mediapipe as mp
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = InceptionResnetV1(pretrained='vggface2').eval().to(device)
@@ -37,4 +38,21 @@ def organize_faces_by_person(embeddings_by_frame, clusters, aligned_faces_folder
         os.makedirs(person_folder, exist_ok=True)
         src = os.path.join(aligned_faces_folder, f"frame_{frame_num}_face.jpg")
         dst = os.path.join(person_folder, f"frame_{frame_num}_face.jpg")
-        shutil.copy(src, dst)
+        shutil.copy(src, dst)
+
+
+def draw_facial_landmarks(image, landmarks):
+    mp_face_mesh = mp.solutions.face_mesh
+    face_mesh = mp_face_mesh.FaceMesh(static_image_mode=False, max_num_faces=1, min_detection_confidence=0.6)
+    mp_drawing = mp.solutions.drawing_utils
+    mp_face_mesh = mp.solutions.face_mesh
+    drawing_spec = mp_drawing.DrawingSpec(thickness=1, circle_radius=1)
+    
+    if landmarks:
+        mp_drawing.draw_landmarks(
+            image=image,
+            landmark_list=landmarks,
+            connections=mp_face_mesh.FACEMESH_TESSELATION,
+            landmark_drawing_spec=drawing_spec,
+            connection_drawing_spec=drawing_spec)
+    return image