san021/phi_ratio
0
1�
2 ���h�3 � � � d dl Z d dlZd dlZd dlZej � ej � e� � ddd� � Z G d� de4� � Z G d� de5� � Z ej
� � Z ej e � � ZnC# e$ r;Z ed e � �� � ed6� � ed e� �� � e� � Y dZ[ndZ[ww xY wd � ZdS )
� Nz..�modelsz%shape_predictor_68_face_landmarks.datc � � e Zd ZdZdS )�NoFaceFoundErrorz>Custom exception raised when no face is detected in the image.N��__name__�7__module__� __qualname__�__doc__� � �!D:\phi_ratio\src\face_detector.pyr r s � � � � � �H�H��Dr r c � � e Zd ZdZdS )�MultipleFacesFoundErrorz9Custom exception raised when multiple faces are detected.Nr r r r
r r s � � � � � �C�C��Dr r zError loading model from zVPlease make sure you have downloaded the file and placed it in the 'models' directory.zError details: c �� � t j | t j � � }t |d� � }t |� � dk rt d� � �t |� � dk rt
d� � �t ||d � � }t j dd�� � }t dd� � D ]9}|� |� � j |� |� � j
f||< �:|S ) a� 8 Detects faces in an image and returns the 68 facial landmarks9 for the first face found.10 11 Args:12 image (numpy.ndarray): The input image (loaded via OpenCV).13 14 Returns:15 numpy.ndarray: A 68x2 NumPy array where each row is an (x, y)16 coordinate of a facial landmark.17 18 Raises:19 NoFaceFoundError: If no faces are detected in the image.20 MultipleFacesFoundError: If more than one face is detected.21 � r z+No face was detected in the provided image.zDMultiple faces were detected. Please provide an image with one face.)�D � �int)�dtyper )�cv2�cvtColor�COLOR_BGR2GRAY�detector�lenr r � predictor�np�zeros�range�part�x�y)�image�gray�rects�shape�coords�is r
�
get_landmarksr( ! s� � �" �<��s� 1� 2� 2�D�
�T�1� � �E� �5�z�z�Q��� � L�M�M�M�22�5�z�z�A�~�~� &�&l�m�m�m�
�d�E�!�H� %� %�E� �X�g�U�
+�
+�
+�F�
�1�b�\�\� 7� 7���Z�Z��]�]�_�e�j�j��m�m�o�6��q� � � �Mr )r �dlib�numpyr �os�path�join�dirname�__file__�23MODEL_PATH� Exceptionr r �get_frontal_face_detectorr �shape_predictorr � RuntimeError�e�print�exitr( r r r
�<module>r8 se �� 24�25�26�27� � � � � � � � � � � � �28 �W�\�\�"�'�/�/�(�3�3�T�8�El�
m�
m�29� � � � � �y� � � � � � � � �i� � � � *�4� )� +� +�� �$��$�Z�0�0�I�I��� � � � �E�302�j�312�322�3�3�3� �E�33b�c�c�c� �E�34 �A�35 �36 � � � ��D�F�F�F�F�F�F�F�F����� ���� +� +� +� +� +s �2B �C�1B>�>C