sOiHzddlmZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZGddZGdd eZGd d eZGd d eZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddZy) )AnyN)Image) NumpyArray) center_cropconvert_to_rgb crop_ndarray normalize pil2ndarrayrescaleresizeresize_longest_edgeresize_ndarray pad2squarecHeZdZdeedeej eezfdZy) Transformimagesreturnctd)Nz%Subclasses must implement this method)NotImplementedError)selfrs j/volume1/homes/robertsu/coba/app/.venv/lib/python3.12/site-packages/fastembed/image/transform/operators.py__call__zTransform.__call__s!"IJJN)__name__ __module__ __qualname__listrrrrrrrrs/KtCyKT%++->jAQ-QKrrcPeZdZdeejdeejfdZy) ConvertToRGBrrc@|Dcgc]}t|c}Scc}w)N)image)rrrr"s rrzConvertToRGB.__call__s9?@U+@@@sN)rrrrrrrrrr r s*AtEKK0AT%++5FArr cReZdZdeeeffdZdeejdeefdZ y) CenterCropsizec||_yNr&)rr&s r__init__zCenterCrop.__init__ s  rrrcV|Dcgc]}t||jc}Scc}w)N)r"r&)rr&r#s rrzCenterCrop.__call__#s%FLMfU %dii8fMMMs&N) rrrtupleintr*rrrrrrrr%r%s:U38_NtEKK0NT*5ENrr%cteZdZdeeezdeeezfdZdeeeeezdeeeeezfdZy) Normalizemeanstdc ||_||_yr(r0r1)rr0r1s rr*zNormalize.__init__(s rrrc ,|rSt|dtr@|Dcgc]2}|Dcgc]$}t||j|j&c}4c}}S|Dcgc]$}t||j|j&c}Scc}wcc}}wcc}w)Nrr3) isinstancerr r0r1rr img_patchesr"s rrzNormalize.__call__,s jD1$*#)KNYY[E5tyydhh?[Y#)  QWWPVuIe$))BPVW W Z Xs B )B B )BB N)rrrfloatrr*rrrrrr/r/'sgUT%[0utE{7J X:&d:.>)?? X j Dj!12 2 Xrr/ceZdZejj fdeeeefzdejfdZde ejde ejfdZ y)Resizer&resamplec ||_||_yr(r&r;rr&r;s rr*zResize.__init__;    rrrcl|Dcgc]$}t||j|j&c}Scc}w)Nr=)r r&r;r#s rrzResize.__call__Cs-SYZSY%u499t}}ESYZZZ)1N) rrrr ResamplingBICUBICr-r,r*rrrrrr:r::se&+%5%5%=%=!E#s(O#!""![tEKK0[T%++5F[rr:cZeZdZddefdZdeeeeezdeeeeezfdZy)Rescalescalec||_yr(rF)rrFs rr*zRescale.__init__Hs  rrrc |rHt|dtr5|Dcgc]'}|Dcgc]}t||jc})c}}S|Dcgc]}t||jc}Scc}wcc}}wcc}w)NrrH)r5rr rFr6s rrzRescale.__call__Ks jD1$*#)K@KK{edjj1{K#)  CII&GE4&I I L Js A5A0 A5A;0A5N)p?)rrrr8r*rrrrrrrErEGsNe J:&d:.>)?? J j Dj!12 2 JrrEcBeZdZdeejezdeefdZy) PILtoNDarrayrrc>|Dcgc] }t|c}Scc}wr()r r#s rrzPILtoNDarray.__call__Zs067u E"777sN)rrrrrrrrrrrLrLYs(8tEKK*$<=8$zBR8rrLcveZdZdedeezeedfzfdZdeejdeejfdZ y) PadtoSquarer& fill_color.c ||_||_yr(r&rP)rr&rPs rr*zPadtoSquare.__init___s  $rrrcl|Dcgc]$}t||j|j&c}Scc}w)N)r"r&rP)rr&rPr#s rrzPadtoSquare.__call__gs6]c ]cTYJUt O]c   rAN) rrrr-strr,r*rrrrrrrOrO^sO%%#Ic3h/% tEKK0 T%++5F rrOceZdZdZej j fdedej fdZde ejde ejfdZ y) ResizeLongestEdgezNResize images so the longest edge equals target size, preserving aspect ratio.r&r;c ||_||_yr(r=r>s rr*zResizeLongestEdge.__init__pr?rrrcj|Dcgc]#}t||j|j%c}Scc}wr()r r&r;r#s rrzResizeLongestEdge.__call__xs.RXYRX#E499dmmDRXYYYs(0N) rrr__doc__rrBLANCZOSr-r*rrrrrrVrVms[X &+%5%5%=%=!!""!ZtEKK0ZT%++5FZrrVc|eZdZdZej j fdedej fdZde e de e fdZ y) ResizeForVisionEncoderz Resize both dimensions to be multiples of vision_encoder_max_size. Preserves aspect ratio approximately. Works on numpy arrays in (C, H, W) format. max_sizer;c ||_||_yr(r]r;rr]r;s rr*zResizeForVisionEncoder.__init__ !   rrrcZg}|D]"}|j\}}}||z }||k\rmtj||jz |jz}t ||z } tj| |jz |jz} nltj||jz |jz} t | |z}tj||jz |jz}t ||| f|j d} |j| %|SNTr&r; channel_first)shapemathceilr]r-rr;append) rrresultr"_heightwidth aspect_ratio new_width new_heightresizeds rrzResizeForVisionEncoder.__call__sE${{ Avu 6>L IIedmm&;N  \ 9:  IIi$--&?@4==P %," G MM' "14 rN rrrrYrrBrZr-r*rrrrrrr\r\|sR&+%5%5%=%=!!""!tJ/D4Drr\ceZdZdZej j fdedej fdZde e de e e fdZ y) ImageSplittera Split images into grid of patches plus a global view. If image dimensions exceed max_size: - Divide into ceil(H/max_size) x ceil(W/max_size) patches - Each patch is cropped from the image - Add a global view (original resized to max_size x max_size) If image is smaller than max_size: - Return single image unchanged Works on numpy arrays in (C, H, W) format. r]r;c ||_||_yr(r_r`s rr*zImageSplitter.__init__rarrrc g}|D]F}|j\}}}|jx}}g} ||kDs||kDrtj||z } tj||z } tj|| z } tj|| z } t | D]\}t | D]L}|| z}|| z}t || z|}t || z|}t |||||d}| j|N^t|||f|jd}| j|n| j||j| I|S)NT)x1y1x2y2rerd) rfr]rgrhrangeminrrirr;)rrrjr"rkrlrm max_height max_widthframes num_splits_h num_splits_woptimal_height optimal_widthrcstart_xstart_yend_xend_ycropped global_views rrzImageSplitter.__call__scE${{ Avu%)]] 2JF "ei&7#yy*)<= #yy):; "&6L+@!A $ %,*> ? |,A"<0"#m"3"#n"4 #Gm$;U C #Gn$s rr*zSquareResize.__init__r?rrrc |Dcgc]2}t||j|jf|jdg4c}Scc}wrc)rr&r;r#s rrzSquareResize.__call__sR    DII 6^b      s7?NrrrrrrrsW&+%5%5%=%=!!""! tJ/ Dj9I4J rrcDeZdZdeefdZdeej eezdeeeej zfdZe de e e fddfdZ edeede e e fddfd Ze deede e e fddfd Zedeede e e fddfd Zedeede e e fddfd Ze deede e e fddfd Zedeede e e fddfdZedeede e e fddfdZedeede e e fddfdZedde dzdej.fdZy)Compose transformsc||_yr(r)rrs rr*zCompose.__init__s $rrrc8|jD] }||} |Sr(r)rr transforms rrzCompose.__call__s!Iv&F) rconfigc8g}|j|||j|||j|||j|||j |||j |||j |||j||||S)aXCreates processor from a config dict. Args: config (dict[str, Any]): Configuration dictionary. Valid keys: - do_resize - resize_mode - size - fill_color - do_center_crop - crop_size - do_rescale - rescale_factor - do_normalize - image_mean - mean - image_std - std - resample - interpolation Valid size keys (nested): - {"height", "width"} - {"shortest_edge"} - {"longest_edge"} Returns: Compose: Image processor. r)_get_convert_to_rgb _get_resize_get_pad2square_get_center_crop_get_pil2ndarray_get_image_splitting _get_rescale_get_normalize)clsrrs r from_configzCompose.from_configs<')   F3  F+ J/ Z0 Z0   V4 V, :v.j))rNc6|jtyr()rir rrs rrzCompose._get_convert_to_rgbB,.)rc |jdd}|dvr|jddrr|d}d|vr|d}nd|vrd |vr |d|d f}n td |jt||jd tj j  yy|d k(rd|vr&d|dvrtd|dj|dd}|jdd}|dkrpt||z }|jt||jd tj j  |jt||fy|jt||f|jd tj j  y|dk(r|jd}t|tr|j|} n|xstj j } d|vr8|jdd} | dk(r |jt|d| yyy|dk(r|jddr|jdi}d|vr td|jd tj j} t| trt j | } |jt|d| yytd|d)Nimage_processor_typeCLIPImageProcessorrSiglipImageProcessor do_resizeFr& shortest_edgerlrmzASize must contain either 'shortest_edge' or 'height' and 'width'.r;r=ConvNextFeatureExtractorz6Size dictionary must contain 'shortest_edge' key. Got crop_pctg?ir)JinaCLIPImageProcessor interpolation resize_modeshortestIdefics3ImageProcessor longest_edgezJSize dictionary must contain 'longest_edge' key for Idefics3ImageProcessor Preprocessor  is not supported)get ValueErrorrir:rrBrCkeysr-r%r5rT_interpolation_resolverrZrV) rrrmoder&rrresize_shortest_edgerr;rs rrzCompose._get_resizeFszz02FG A Azz+u-f~"d*0D%'T/ NDM:D$[!!!!'J8H8H8P8P!Q. / /O6&>$I LVTZ^M`M`MbLcd#6N?;Mzz*e4Hs"'*=8+C'D$!!1!'J8H8H8P8P!Q !!*=-2P"QR!!+];!'J8H8H8P8P!Q - -"JJ7M--66}E(DE,<,<,D,D$jj C *,%%!'%-- - -zz+u-zz&"-!-$d "::j%2B2B2J2JKh,$//9H!!%!.1!).&}TF2CDE ErcT|jdd}|dvrr|jddr_|d}t|tr||f}n)t|tr |d|df}nt d ||j t | yy|d k(ry|d k(ry|d k(ryt d|d)Nrrrdo_center_cropF crop_sizerlrmzInvalid crop size: r)rrrrr)rr5r-dictrrir%)rrr crop_size_rawrs rrzCompose._get_center_cropszz02FG A Azz*E2 &{ 3 mS1!. >I t4!.x!8-:P QI$':=/%JKK!!*)"<=3/ /  - -  - - }TF2CDE Erc6|jtyr()rirLrs rrzCompose._get_pil2ndarrayrrc|jdd}|dk(r|jdd}|jdijdd}|jd tjj}t |t rtj|}|r7|j t|||j t||y |j t||y y ) z Add image splitting transforms for Idefics3. Handles conditional logic: splitting vs square resize. Must be called AFTER PILtoNDarray. rrrdo_image_splittingFmax_image_sizerir;N) rrrBrZr5r-rir\rtr)rrrr do_splittingr]r;s rrzCompose._get_image_splittingszz02FG + +!::&:EBLzz"2B7;;NCPHzz*e.>.>.F.FGH(C( ++H5!!"88"LM!!-("CD!!,x"BC ,rc|jddr.|jdd}|jt|yy)N do_rescaleTrescale_factorrJrH)rrirE)rrrs rrzCompose._get_rescales9 ::lD )#ZZ(8'BN   gN; < *rc|jddr#|jt|d|dyd|vr(d|vr#|jt|d|dyyy)N do_normalizeF image_mean image_stdr3r0r1)rrir/rs rrzCompose._get_normalizesb ::ne ,   iVL-AvkGZ[ \ v %6/   iVF^O P#2 rc |jdd}|dk(ry|dk(ry|dk(r0|jt|d|jddyy) Nrrrrr&rPrrR)rrirO)rrrs rrzCompose._get_pad2squaresezz02FG ' '  / /  - -   %zz,: .rr;ctjjtjjtjjtjj tjj tjjd}|r#|j|jx}r|Std|)N)nearestlanczosbilinearbicubicboxhammingzUnknown interpolation method: ) rrBNEARESTrZBILINEARrCBOXHAMMINGrlowerr)r;interpolation_mapmethods rrzCompose._interpolation_resolvers''//''//((11''//##''''//   #4#8#89I#JJJM9(DEErr()rrrrrr*rrr classmethodrrTrr staticmethodrrrrrrrrrBrrrrrrsq%4 ?%5;;'$z*:: j D- -&*c3h&*I&*&*P*Y*c3h*TX**OFT)_OFd38nOFQUOFOFbFT)_Fd38nFQUFF,*T)_*d38n*QU**Dd9oDtCQTH~DZ^DD*=i=$sCx.=T== Q4 ?QDcNQtQQ  DO T#s(^ PT   F#* F@P@P F Frr)typingrrgPILrfastembed.common.typesr$fastembed.image.transform.functionalrrrr r r r r rrrr r%r/r:rErLrOrVr\rtrrrrrrs -   KK A9A NNX X& [Y [JiJ$898  )   Z Z+Y+\HIHV 9 2eFeFr