diff --git a/face_alignment/api.py b/face_alignment/api.py index ee896a09..62d3031a 100644 --- a/face_alignment/api.py +++ b/face_alignment/api.py @@ -181,7 +181,7 @@ def get_landmarks_from_image(self, image_or_path, detected_faces=None): landmarks.append(pts_img.numpy()) - return landmarks + return detected_faces,landmarks def get_landmarks_from_directory(self, path, extensions=['.jpg', '.png'], recursive=True, show_progress_bar=True): detected_faces = self.face_detector.detect_from_directory(path, extensions, recursive, show_progress_bar)