|hY%ddlZddlZddlZddlmZmZmZmZdejd<ddl Z ddl Z ddl mZddlmZddlmZmZddlmZdd lmZej.ed Zej2d d ejd<dejd<ee jj7dddZedk(r1e j:j=ej>ddl m!Z!nddl"m!Z!defdZ# d dee$e jJfde&dee!de'dee$eff dZ( d!dee$e jJfde&dee!de'de'dede)fdZ*y)"N)UnionAnyOptionalDict1TF_USE_LEGACY_KERAS) __version__)retinaface_model) preprocess postprocess)Logger) package_utilszretinaface/RetinaFace.py)moduleignore3TF_CPP_MIN_LOG_LEVELtrueTF_FORCE_GPU_ALLOW_GROWTH.)maxsplit)ModelreturncdtvrOtjtjtj gdt jfatS)z? Builds retinaface model once and store it into memory model)NNN)shapedtype)input_signature) globalstffunctionr build_model TensorSpecnpfloat32rT/var/www/html/test/engine/venv/lib/python3.12/site-packages/retinaface/RetinaFace.pyr$r$-sE gi   ( ( *]]1FbjjY[  Lr)img_path thresholdrallow_upscalingc i}tj|}| t}d}d}gd}tjgdgdgtj tjgdgd gtj tjgd gd gtj d } d d d d } g} g} g} tj ||\}}}||}|Dcgc]}|j}}d}t|D]]\}}||}|dddddd| d|df}||dz}|jd|jd }}| d|}||z}| d|}tj||||}|j||zdf}|jd}gd}|jd|z}|jd|f}|dddddf|dz|dddddf<|dddddf|dz|dddddf<|ddd ddf|d z|ddd ddf<|dddddf|dz|dddddf<tj||} tj| |dd } |dk(r |dkr||z}|j}!tj |!|k\d}"| |"ddf} ||"}| ddddfxx|zcc<| j#| | j#|||d z}#|#jd|z}$|#jdd|$dzf}#tj$||#}%|%|"ddf}%|%dddddd fxx|zcc<| j#|%|dz }`tj&| } | jddk(r|Stj&| }|j}!|!j)ddd}"| |"ddf} ||"}tj&| }%|%|"j+tj d}%tj,| ddddf|fj+tj d}&tj.|&|}'tj,|&| ddddff}(|(|'ddf}(|%|'}%t|(D]\})}*dt1|)dzz}+i||+<|*d||+d<t3|*ddj+t4||+d<i||+d<t3|%|)d||+dd<t3|%|)d||+dd<t3|%|)d ||+dd <t3|%|)d||+dd!<t3|%|)d||+dd"<|Scc}w)#a Detect the facial area for a given image Args: img_path (str or numpy array): given image threshold (float): threshold for detection model (Model): pre-trained model can be given allow_upscaling (bool): allowing up-scaling Returns: detected faces as: { "face_1": { "score": 0.9993440508842468, "facial_area": [155, 81, 434, 443], "landmarks": { "right_eye": [257.82974, 209.64787], "left_eye": [374.93427, 251.78687], "nose": [303.4773, 299.91144], "mouth_right": [228.37329, 338.73193], "mouth_left": [320.21982, 374.58798] } } } Ng?g?) )or2pp@r3)^r4`@r5)r)Lr6Q@r7)8r8C@r9) r:7@r;)r<.@r=)stride32stride16stride8rrstrider)r)?rDrDrDrrCrDF)copyface_score facial_area landmarks right_eyeleft_eyenose mouth_right mouth_left)r get_imager$r&arrayr'preprocess_imagenumpy enumeraterr anchors_planereshape bbox_pred clip_boxesravelwhereappend landmark_predvstackargsortastypehstackcpu_nmsstrlistint),r+r,rr-respimg nms_thresholddecay4_feat_stride_fpn _anchors_fpn _num_anchorsproposals_list scores_listlandmarks_list im_tensorim_infoim_scalenet_outeltsym_idx_sscores bbox_deltasheightwidthAK anchors_fpnanchors bbox_stds bbox_pred_len proposals scores_ravelorderlandmark_deltaslandmark_pred_lenrJpre_detkeepdetidxfacelabels, r* detect_facesr=s0: D   x (C } M F"HH +-K LTVT^T^ HH ')C DBJJ 8857MNVXV`V`aL!"qQ?LNKN#-#>#>sO#T IwIG&-.ssyy{.G.G*+.1!1aqcl!;!==>gk* #))!,k.?.?.B 6! & UN"VA3<0 ++FE1kJ//1q5!*-(( #))!,1 !))2}*=> *1add73ilB Aqt!tG*1add73ilB Aqt!tG*1add73ilB Aqt!tG*1add73ilB Aqt!tG))';? **9gbqkB 6fsl f F||~ 23A6eQh' !QqS&X%i(6"!'A+.+11!49)112q:Kq:P2QR--gG eQh' !Q!)(i(1 ].` .)IqQ YY{ #F<<>L  "4R4 (E%(#I E]F .)I% '' '?Iii1ac6*F34;;BJJU;SG   w 6D ))Wi12./ 0C dAg,C$Is^ I T#cAg,&U #AwU G%)$q)*:*:3*?%@U M"#%U K 04Ys^A5F0GU K -/3IcN14E/FU K ,+/ #q0A+BU K (26y~a7H2IU K /15inQ6G1HU K . I Ks/sU;alignexpand_face_areac g}tj|}t||||}t|ts|S|j D]\} } | d} d} d} | d}| d}| d|z }| d|z }|dkDr|t ||zdz z}|t ||zdz z}td|t ||z dz z }td|t ||z dz z }t|jd|z |}t|jd|z |}||||z|||zf}|dur| d }|d }|d }|d }tj|||| \}} } tj||||z||zf| | |jd|jdf\}}}}|t |t |t |t |f}|j|dddddddf|S)a Extract detected and aligned faces Args: img_path (str or numpy): given image threshold (float): detection threshold model (Model): pre-trained model can be passed to the function align (bool): enable or disable alignment allow_upscaling (bool): allowing up-scaling expand_face_area (int): expand detected facial area with a percentage )r+r,rr-rIrrrrdTrJrLrKrM)rfrLrKrMNrC)r rPr isinstancedictitemsrdmaxminrr alignment_procedurerotate_facial_arear[)r+r,rrr-rrerfobjruidentityrI rotate_anglerotate_directionxywh expanded_w expanded_h facial_imgrJrLrKrM aligned_img rotated_x1 rotated_y1 rotated_x2 rotated_y2s r* extract_facesrsc$ D   x (C   C c4  yy{+, 8}-   N N NQ  NQ  a S%5!5!;<I=[=[Aq1ua!e$l4DsyyQR|UXU^U^_`UaFb> :J J %J#j/13z?S_3TTJ Jq!TrTz*+W+,Z Kr))?NT)rNTTr)+oswarningsloggingtypingrrrrenvironrSr& tensorflowr" retinafacer retinaface.modelr retinaface.commonsr r retinaface.commons.loggerr rvalidate_for_keras3loggerfilterwarningsrdsplit tf_version get_loggersetLevelERRORtensorflow.keras.modelsr keras.modelsr$rbndarrayfloatboolrrcrr(r)r*rs --%( !"-6,," !!# 1 2!%( !"*0 &' %%cA%6q9 :  ?BMMOW]]+-" S $! XCO$XX E?X X  #s(^ Xz!  NCO$NN E?N  N  N  N Nr)