Update utils.py
Browse files
utils.py
CHANGED
|
@@ -60,6 +60,10 @@ def create_annotated_video(video_path, df, mse_embeddings, largest_cluster, outp
|
|
| 60 |
box = boxes[0]
|
| 61 |
cv2.rectangle(frame, (int(box[0]), int(box[1])), (int(box[2]), int(box[3])), (0, 255, 0), 2)
|
| 62 |
# Draw facial landmarks
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
face_mesh_results = face_mesh.process(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
|
| 64 |
if face_mesh_results.multi_face_landmarks:
|
| 65 |
for face_landmarks in face_mesh_results.multi_face_landmarks:
|
|
|
|
| 60 |
box = boxes[0]
|
| 61 |
cv2.rectangle(frame, (int(box[0]), int(box[1])), (int(box[2]), int(box[3])), (0, 255, 0), 2)
|
| 62 |
# Draw facial landmarks
|
| 63 |
+
mp_face_mesh = mp.solutions.face_mesh
|
| 64 |
+
mp_drawing = mp.solutions.drawing_utils
|
| 65 |
+
mp_drawing_styles = mp.solutions.drawing_styles
|
| 66 |
+
face_mesh = mp_face_mesh.FaceMesh(static_image_mode=False, max_num_faces=1, min_detection_confidence=0.5)
|
| 67 |
face_mesh_results = face_mesh.process(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
|
| 68 |
if face_mesh_results.multi_face_landmarks:
|
| 69 |
for face_landmarks in face_mesh_results.multi_face_landmarks:
|