Explicite/api-test
0
1�
2 �_�i_ � � � d dl Zd dlZd dlmZ dad� Zd� Zd� Zdej dej de3fd �Z d4ej d e dej fd �Z
y)
� N)� FaceAnalysisc �` � t �#t d�� a t j dd�� t S )N� buffalo_l)�name�����)� r )�ctx_id�det_size)� _face_appr �prepare� � �^/Users/alexiseduardojuarezjaimes/Desktop/api-face-extractor/app/services/insightface_engine.py� get_face_appr s+ � ��� �k�2� �����j��9� �r c �( � | syd� }t | |�� S )uL 5 Si hay varias caras, escogemos la de mayor área (lo más común).6 Nc � � | j j t � \ }}}}t d||z7 � t d||z8 � z S )Nr )�bbox�astype�int�max��face�x1�y1�x2�y2s r �areaz pick_best_face.<locals>.area s= � � ���)�)�#�.���B��B��1�b��e�}�s�1�b��e�}�,�,r )�key)r )�facesr s r �pick_best_facer s � � ��-� �u�$� � r c � � | j j t � \ }}}}t |� t |� t |� t |� d�S )N)�top�right�bottom�left)r r r r s r � to_bbox_dictr& s= � ��Y�Y�%�%�c�*�N�B��B���r�7�S��W��B���R�� Q�Qr �a�b�returnc � � | j d� } |j d� }| t j j | � dz z } |t j j |� dz z }t t j9 | |� � S )N�float32g�-���q=)r �np�linalg�norm�float�dot)r' r( s r �cosine_similarityr1 ! sk � � �����A� �����A� �R�Y�Y�^�^�A�
��
&�'�A� �R�Y�Y�^�^�A�
��
&�'�A� �����1� � � r �np_rgb�seedc � � t j j |� }| j � }t |j dd� � }t |j dd� � }t
j |||�� }|j dd \ }}t |j dd � � }t
j |dz |dz f|d10� } t
j || ||ft j t j � � }|S ) ur 11 Variantes suaves: brillo/contraste + rotación leve.12 Esto ayuda cuando hay barba, luz distinta, etc.13 g�������?g�������?i����� )�alpha�betaN� g �g @g �?)�flags�14borderMode)
r, �random� default_rng�copyr/ �uniform�cv2�convertScaleAbs�shape�getRotationMatrix2D�15warpAffine� INTER_LINEAR�BORDER_REFLECT)16r2 r3 �rng�imgr6 r7 �h�w�angle�Ms17 r � augment_rgbrL ( s� � �18
�)�)�19 �20 ��21%�C�22�+�+�-�C�
�#�+�+�d�D�)� *�E� ����S�"�%� &�D�23
�24 �25 �c��T�26:�C� �9�9�R�a�=�D�A�q� �#�+�+�d�C�(� )�E� � � ��Q���A����s�;�A�27
�.�.��a�!�Q��s�/?�/?�C�L^�L^�28_�C� �Jr )�numpyr, r? �insightface.appr r r r r& �ndarrayr/ r1 r rL r
r r �<module>rP sg �� � 29� (� � �� �R�30 ���� ��31�32� �u� ���33�34� �#� �"�*�*� r