|hddlZddlZddlmZddlmZddlZddlZddl Z dee ejfdejfdZ dejfdZ de dejfd Zd ejd ed edefd Zd ejd edefdZy)N)Path)Unionimg_urireturnct|tjr|j}nt|tr|j dr t |}nt|tr|j dr t|}nt|ttfr_t|tr t |}tjj|std|dtj|}ntd|dt|j dk7s"tj"|j d k(r td |S) z Load the given image Args: img_path (str or numpy array): exact image path, pre-loaded numpy array (BGR format) , base64 encoded string and urls are welcome Returns: image itself z data:image/http)urlzInput image file path (z) does not exist.zInvalid image input - zS.Exact paths, pre-loaded numpy arrays, base64 encoded strings and urls are welcome.rz:Input image needs to have 3 channels at must not be empty.) isinstancenpndarraycopystr startswithload_base64_imgload_image_from_webrospathisfile ValueErrorcv2imreadlenshapeprod)rimgs \/var/www/html/test/engine/venv/lib/python3.12/site-packages/retinaface/commons/preprocess.py get_imager s'2::&lln GS !g&8&8&Gg& GS !g&8&8&@!g. Gc4[ ) gt $'lGww~~g&6wi?PQR Rjj!$WI., ,   399~bggcii0A5UVV Jc|jdd}tjtj|tj }t j|t j}|S)zLoad image from base64 string. Args: uri: a base64 string. Returns: numpy array: the loaded image. ,) splitr fromstringbase64 b64decodeuint8rimdecode IMREAD_COLOR)uri encoded_datanparrimg_bgrs rrr8sP99S>!$L MM&**<8"(( CEll5#"2"23G Nrr c*tj|dd}|jtjt |j jtj}tj|tj}|S)z Loading an image from web Args: url: link for the image Returns: img (np.ndarray): equivalent to pre-loaded image from opencv (BGR format) T<)streamtimeoutdtype) requestsgetraise_for_statusr asarray bytearrayrawreadr'rr(r))r response image_arrayimages rrrHsc||Cb9H **Yx||'8'8':;288LK LLc&6&6 7E Lrrscalesallow_upscalingcD|jdd\}}|d}|d}||kDr||}}n||}}|t|z } |s td| } tj| |z|kDr|t|z } | dk7r)t j |dd| | t j}|| fS)aS This function is modified from the following code snippet: https://github.com/StanislasBertrand/RetinaFace-tf2/blob/5f68ce8130889384cb8aca937a270cea4ef2d020/retinaface.py#L49-L74 Args: img (numpy array): given image scales (list) allow_upscaling (bool) Returns resized image, im_scale rr"?N)fxfy interpolation)rfloatminr roundrresize INTER_LINEAR) rr>r?img_himg_w target_sizemax_size im_size_min im_size_maxim_scales r resize_imagerRWs99Qq>LE5)KayH u}#(%[ #(%[ U;//H sH% xx;&'(2eK003jjdDX(RURbRbc =rc\tjgdtj}tjgdtj}td}ddg}t |||\}}|j tj}tj d|jd|jd|jd ftj}td D]1}|d d d d d |z f|z |d |z z |d |z z |dd d d d |f<3||jdd |fS) aC This function is modified from the following code snippet: https://github.com/StanislasBertrand/RetinaFace-tf2/blob/5f68ce8130889384cb8aca937a270cea4ef2d020/retinaface.py#L76-L96 Args: img (numpy array): given image allow_upscaling (bool) Returns: tensor, image shape, im_scale )rTrTr2)rBrBrBrBiir"rrAr N) r arrayfloat32rFrRastypezerosrrange) rr? pixel_means pixel_stds pixel_scaler>rQ im_tensoris rpreprocess_imager_ys((?"**=K/MC **RZZ C!SYYq\399Q<1FbjjYI1X !$Q1q5[!1K!?+aRSeBT!TXb EY ! !Q1* cii!nh ..r)rr%pathlibrtypingrr4numpyr rrr rrrlistbooltuplerRr_rrrgs  +uS"**_-+"**+\ BJJ  S RZZ bjj$%D/"**/t//r