ihz ddlmZmZmZddlZddlmZddlm Z m Z m Z ddl m Z mZmZmZmZmZmZmZmZddlmZedZd ed efd Z dd ed ej6d eded ee f dZdeeeeefdedeeefd eeeeeffdZ y))AnyListTupleN) detection)Detector DetectedFaceFacialAreaRegion) FastMtCnn MediaPipeMtCnnOpenCvDlib RetinaFaceSsdYoloYuNet)Loggerz%deepface/detectors/DetectorWrapper.py)moduledetector_backendreturnc tjtjtj t jtjtjtjtjt j"d }dt%vriat)t&j+}||vr:|j-|}|r|}|t&|<t&|St/d|zt&|S)z Build a face detector model Args: detector_backend (str): backend detector name Returns: built detector (Any) ) opencvmtcnnssddlib retinaface mediapipeyolov8yunet fastmtcnnface_detector_objz"invalid detector_backend passed - )r OpenCvClientr MtCnnClientr SsdClientr DlibClientrRetinaFaceClientr MediaPipeClientr YoloClientr YuNetClientr FastMtCnnClientglobalsr!listkeysget ValueError)rbackends built_models face_detectors `/var/www/html/dev/engine/venv/lib/python3.12/site-packages/deepface/detectors/DetectorWrapper.py build_modelr4s%%""}} 11..//"".. H ') +)..01L|+ %56 )OM2? . / - ..ADTTU U - ..imgalignexpand_percentagec Vt|}|dkrtjd|dd}|j|}g}|D]}|j}|j } |j } |j} |j} |j} |j}|dkDr| t| |zdz z}| t| |zdz z}td|t|| z dz z }td| t|| z dz z } t|jd|z |} t|jd| z |} |t| t| | zt|t|| zf}|durtj || | \}}t#|| || z| | zf||jd|jdf \}}}}|t|t|t|t|f}t%|t'|| | | || | | }|j)||S) a Detect face(s) from a given image Args: detector_backend (str): detector name img (np.ndarray): pre-loaded image align (bool): enable or disable alignment after detection expand_percentage (int): expand detected facial area with a percentage (default is 0). Returns: results (List[DetectedFace]): A list of DetectedFace objects where each object contains: - img (np.ndarray): The detected face as a NumPy array. - facial_area (FacialAreaRegion): The facial area region represented as x, y, w, h - confidence (float): The confidence score associated with the detected face. rz7Expand percentage cannot be negative but you set it to z.Overwritten it to 0.)r6dT)r6left_eye right_eye) facial_areaanglesize)xyhw confidencer=r>)r6r?rF)r4loggerwarn detect_facesrBrCrErDr=r>rFintmaxminshaper align_facerotate_facial_arearr append)rr6r7r8r2 facial_areasresultsr?rBrCrErDr=r>rF expanded_w expanded_h detected_face aligned_imgr@ rotated_x1 rotated_y1 rotated_x2 rotated_y2results r3rIrI;sn0**:;M1 EFWEXY# # !--#-6LG#, MM MM MM MM''))  ++ q S%6!6!<==JS%6!6!<==JAq3 Q!3445AAq3 Q!3445ACIIaL1$j1ACIIaL1$j1ACFSQZ/Q#a!e*1DDE  D=!*!5!5(i" K>P1q5!a%0iilCIIaL1> :J J (J#j/1J#j/123M(qAzHXa"   vY,Z Nr5r?r@rAc|dk\rdnd}t|dz}|dk(r|S|tjzdz }|d|dzdz |ddz z }|d|dzdz |ddz z }|tj|z||ztj|zz}| |ztj|z|tj|zz}||ddz z}||ddz z}||d|dz dz z }||d|dz dz z } ||d|dz dz z} ||d|dz dz z} t |t | t | t | fS)a Rotate the facial area around its center. Inspried from the work of @UmutDeniz26 - github.com/serengil/retinaface/pull/80 Args: facial_area (tuple of int): Representing the (x1, y1, x2, y2) of the facial area. x2 is equal to x1 + w1, and y2 is equal to y1 + h1 angle (float): Angle of rotation in degrees. Its sign determines the direction of rotation. Note that angles > 360 degrees are normalized to the range [0, 360). size (tuple of int): Tuple representing the size of the image (width, height). Returns: rotated_coordinates (tuple of int): Representing the new coordinates (x1, y1, x2, y2) or (x1, y1, x1+w1, y1+h1) of the rotated facial area. rr<ihr;)absnppicossinrJ) r?r@rA directionrBrCx_newy_newx1y1x2y2s r3rOrOs2aZRI J E z BEEMC E Q+a. (A-Q! ;A Q+a. (A-Q! ;A u I u = =E BNRVVE] *Q-> >E DGaK E DGaK E +a.;q>1Q6 6B +a.;q>1Q6 6B +a.;q>1Q6 6B +a.;q>1Q6 6B GSWc"gs2w //r5)Tr)!typingrrrnumpyradeepface.modulesrdeepface.models.Detectorrrr deepface.detectorsr r r r rrrrrdeepface.commons.loggerrrGstrr4ndarrayboolrJrIfloatrOr5r3rws##&MM   + > ?#/##/##/NZ[SS " S37SSVS ,Sl30sCc)*30 30 S/30 3S#  30r5