Update extract_kp_videos.py

Zz-ww · Sep 17, 2023 · f147a90 · f147a90
1 parent 131466e
commit f147a90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/face3d/extract_kp_videos.py b/src/face3d/extract_kp_videos.py
@@ -13,7 +13,7 @@
 
 class KeypointExtractor():
     def __init__(self, device):
-        self.detector = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, device=device)   
+        self.detector = face_alignment.FaceAlignment(face_alignment.LandmarksType.TWO_D, device=device)   
 
     def extract_keypoint(self, images, name=None, info=True):
         if isinstance(images, list):