^jwtddlmZddlmZddlmZmZmZddlZddl Z ddl Z ddl ZddlmZmZddlmZmZmZmZmZmZmZmZmZmZe j:e j<e j>dZ ej:ej<ejBdZ"dEdZ#dFd Z$dGd Z%dGd Z& dHd Z'edIdJd Z( dKdZ) dLdZ* dMdZ+ dNdZ, dOdZ-dPdZ.edQdZ/dRdZ0edSdZ1edTdZ2edFdZ3dUdVdZedUdPdZ4dRdZ5dPdZ6edUdSdZ7edWdZ8edUdWdZ9dUdVd ZdXd!Z:edXd"Z;edXd#ZedGd&Z?dUd[d'Z@edUd\d(ZdEd)ZAedEd*ZBe dU d]d+ZCedEd,ZDd^d-ZEed^d.ZFd_d/ZGedUd_d0ZHed`d1ZIed`d2ZJed`d3ZKd`d4ZLdadbd5ZMedadbd6ZNedadbd7ZOdadbd8ZPdadcd9ZQedadcd:ZRdadcd;ZSeddd<ZTeddd=ZUddd>ZVeddd?ZWeddd@ZXdddAZYdedBZZdfdCZ[dfdDZ\y)g) annotationswraps)AnyCallableLiteralN) contiguouspreserve_channel_dim) MAX_OPENCV_WORKING_CHANNELSMAX_VALUES_BY_DTYPEMONO_CHANNEL_DIMENSIONSNormalizationType ValueTypeclipclipped convert_value get_max_valueget_num_channels)multiplyaddpowerc f|j}|j}|j|k7rt|j|d|d}t j t j|jd|j|djd|||j|S)NFcopyTinplace)alphabetadtype) shaper!rastypenp frombuffersswsumreshape)img1weight1img2weight2original_shapeoriginal_dtypes ]/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/albucore/functions.pyadd_weighted_simsimdr0sZZNZZN zz^#DKKUK;^UYZ ==  R $++n5+"I"Q"QRT"U]dkrs gct|d|dSN)r0imgvalues r/add_array_simsimdr8-s Qq 11r1cDt||tj|dSNr)r0r$ zeros_liker5s r/multiply_by_constant_simsimdr<1s UBMM#,> BBr1ct|dtj||zj|jddS)Nr4Fr)r0r$ ones_liker#r!r5s r/add_constant_simsimdr?5s7 Qc):U)B(J(J399[`(J(acd eer1ctt|}|tjk(r|dk(rtj|}tj|tj j dd}tjd|dztj }|tvrt|||Std|)Nrr rr4rzUnsupported operation: ) r r$uint8truncarrayfloat32r(arange np_operations ValueError)r!r7 operation max_valueluts r/create_lut_arrayrK9s $E*I Y%/ HHU"** - 5 5b! > Lr1c|jtjk(r!|jtjd}|j dk(r|j ddd}tj||j}|dk(r|jtjk(rhtj|dk\rt|tjdStj|jtjdS|S)NFrr4rrrr) r!r$float64r#rDndimr( broadcast_tor"rAallrrBras r/r`r`s  {{bjj  RZZe 4 zzQ aB' OOE399 -EEcii2883 66%1* rxx7 7xx%%bjju%== Lr1c|dk(r2|jtjk(rtj|}t ||j tj d|S)NrFr)r!r$rAint16rFr#rDras r/ apply_numpyrmsK Ecii2883  #CJJrzzJ$F NNr1ct||d|SNrr\r6r7rs r/ multiply_lutrrs S%W 55r1ct||d}|jtjk(r5t j |j tjd|St j ||S)NrFr)rbr!r$rArVrr#rDr5s r/multiply_opencvrtsS eZ 8E yyBHH||CJJrzzJ>FF <<U ##r1ct||dSrormr5s r/multiply_numpyrws sE: ..r1c|jtjk(r t|||S|jtjk(r t ||St ||Sr^)r!r$rArrrDrwrtrqs r/multiply_by_constantrysK yyBHHC00 yyBJJc5)) 3 &&r1c|jtjk(r t|||S|tkDr t ||St ||Sr^)r!r$rArrr rwrt)r6r7rYrs r/multiply_by_vectorr{sF yyBHHC0011c5)) 3 &&r1ct||Sr^)rtr5s r/multiply_by_arrayr}s 3 &&r1ct|}t||}t|ttfr t |||St|t jr|jdk(rt||||St||Sr3) rrrSrUrTryr$ndarrayrhr{r}r6r7rrYs r/rrsk#C(L % .E%%&#C88%$q!#ulGDD S% ((r1c*t||d}|jtjk(xrXt |t t fxr|dkxs9t |tjxr|jtjk7}|rptj|jtjdt |t t fr|S|jtjdS|r|nd}tj|||S)NrrFr)dst) rbr!r$rArSrTrUrrVrr#rD)r6r7r needs_floatrs r/ add_opencvrs eU 3E ))rxx' EC< ( 6UQYvJubjjQRcggc!Q'lDG4R SSRs*E5ct|}t||}t||}|jtjk(r t |||St |||Sr^)rrr!r$rArr)r6rrrYs r/ normalizerEsR#C(L \:K | ,D yyBHHS$ 44 C{ 33r1ct||dSNrrv)r6exponents r/ power_numpyrOs sHg ..r1c|jtjk(rtj||S|jtj k(r$t ||k(rtj||S|jtj k(rGt|tr7|jtjd}tj||Std|jd|)z(Handle the 'power' operation for OpenCV.FrzUnsupported image type z for power operation with value ) r!r$rDrVpowrArTrSrUr#rG)r6r7 img_floats r/ power_opencvrSs yyBJJwwsE"" yyBHHUu!4wwsE"" yyBHHE5!9JJrzzJ6 wwy%(( .syyk9YZ_Y`a bbr1ct||d|Srrp)r6rrs r/ power_lutres S(GW 55r1ct|}t||}|jtjk(r t |||St |ttfr t||St||Sr^) rrr!r$rArrSrUrTrr)r6rrrYs r/rris_#C(LX|4H yyBHHh00(UCL)C** sH %%r1c|jtjd|z|jtjd|zzSrrr)r*r+r,s r/add_weighted_numpyrvs; ;;rzz; . 84;;rzzX];;^ah;h hhr1c4tj||||dSr:)rV addWeightedrs r/add_weighted_opencvrzs ??4$ ;;r1c|j}t|}|dk(r|dk(r|S|dk(r|dk(r|S|dk(r|dk(rtj|S|dk(r|dk(r t |||Stj d|dztj |z}tj||}tj d|dztj |z} tj|| } t|| |S)Nr4rr ) r!r r$r;rrErDrVrr) r)r*r+r,rr!rIlut1result1lut2result2s r/add_weighted_lutrs JJE#E*I!|1  !|1  !|1 }}T""!|1 tW-- 99Q A RZZ 87 BDggdD!G 99Q A RZZ 87 BDggdD!G gw 00r1c|j|jk7r&td|jd|jdt||||S)Nz/The input images must have the same shape. Got z and .)r"rGr0rs r/ add_weightedrsL zzTZZJ4::,V[\`\f\f[gghijj gtW ==r1cJt|ttfrA|dk(rCV_64Fr!rrs r/multiply_add_opencvrs%#u&5A:*VcSX\:Z_eij_j}}S!! ZZ Z /F Q; VR\\&1F:#**M ]]6 3 aZ6dSWWVR\\&5IE5QY\YcYc%ddr1c |j}t|}t|}t|tt fr^t|tt frHt tjd|dztj|z|z|d}t|||St|tjr!|jdk7r|jdd}t|tjr!|jdk7r|jdd}t tjd|dztj|z|z|d}tjt!|D cgc]} t|dddd| f|| |c} Scc} w) Nrr4r FrrrT)r!r rrSrUrTrr$rErDrQrr"r(rVrWrX) r6rr7rr!rIrYrJrZr[s r/multiply_add_lutrs2 IIE#E*I#C(L&5#,'Juucl,K299Q A RZZ@6IEQSXbghc3((&"**%&,,"*<A&%$): b!$  !Y]"**=FNPU_c dD 99eLFYZfSAq\47G<Z [[Zs"!F ct|}t||}t||}|jtjk(rt ||||St |||Sr^)rrr!r$rArr)r6rr7rrYs r/ multiply_addrsT#C(L 6< 0F % .E yyBHHVUG<< sFE 22r1cj|jtjd}d}|jtk(rtj |d}|dk(s|j ddk(r|dk(r|jj}|jj|z}|j dtkDr,tj||}tj||}tjtj|||}tj|d d | S|dk(rtj |\}}|ddd f}|ddd f}|j dtkDr,tj||}tj||}tjtj|||tj" }tj|d d | S|dk(s|j ddk(r\|dk(rW|j%}|j'}tj(|dd dtj*tj"S|dk(r|j%d}|j'd}|j dtkDr,tj||}tj||}tjtjtj||||z |ztj" d d | St-d|)NFr-C6?raxisimager4image_per_channeloutrr min_maxmin_max_per_channel)rr norm_typer!rr4Unknown normalization method: )r#r$rDrhr expand_dimsr"ritemstdr full_likerVdividerr meanStdDevrminmaxr NORM_MINMAXrG)r6 normalizationepsrrnormalized_imgimg_minimg_maxs r/normalize_per_image_opencvrs **RZZe* ,C C xx**nnSr*CIIbMQ$6=L_;_xxz gginn$ 99R=6 6<<T*D,,sC(CCLLd$;SAww~sBNCC++NN3' cAqDz!Q$i 99R=6 6<<T*D,,sC(CCLLd$;S Sww~sBNCC !ciimq&8]Nc=c'')'')}}S$aa3??Z]ZdZdee--''v'&''v'& 99R=6 6ll30Gll30Gww JJs||C1Gg4E4KTWT^T^ _      5m_E FFr1c|jtjd}d}|jtk(rtj |d}|dk(rD|j }|j|z}||z |z }tj|dd| S|d k(rH|j d }|jd |z}||z |z }tj|dd| S|d k(rE|j}|j} tj||z | |z |zz dd| S|d k(rI|jd }|jd } tj||z | |z |zz dd| Std|)NFrrrrrrrrrrrrr) r#r$rDrhr rrrrrrrG) r6rrrrr pixel_mean pixel_stdrrs r/normalize_per_image_numpyrs} **RZZe* ,C C xx**nnSr*xxzggi#o*+ww~sBNCC++XX6X* GGG(3.  *i7ww~sBNCC !'')'')wwg 'G*;c*ABCQTUU--''v'&''v'&wwg 'G*;c*ABCQTUU 5m_E FFr1c p|j}t|}d}t|}|jtk(rt j |d}|dk(s|jddk(r}|dk(rx|j}|j|z}t jd|dzt j|z |z }tj||jd d S|dk(r|jd } |jd |z} t|D cgc]7} t jd|dzt j| | z | | z 9} } tj t|D cgc]5} tj|dddd| f| | jd d 7c} S|d k(s|jddk(r|d k(r{|j#}|j%}t jd|dzt j|z ||z |zz }tj||jd d S|d k(r|j#d }|j%d }t|D cgc]@} t jd|dzt j|| z || || z |zz B} } tj t|D cgc]%} tj|dddd| f| | 'c} St'd|cc} wcc} wcc} wcc} w)Nrrrrr4rrr rrrrrr)r!r rrhr r$rr"rrrErDrVrrrXrWrrrG)r6rr!rIrrYrrrJrrcrZr[rrs r/normalize_per_image_lutr3s IIE#E*I C#C(L xx**nnSr*CIIbMQ$6=L_;_xxzggi#oyyIM> Cy 99r1c|ddddddfSNr.rr6s r/ hflip_numpyr s q$B$| r1c|jdkDr|jddkDr t|dStj|dS)Nr4 flip_coderhr"_flip_multichannelrVflipr s r/ hflip_cv2r< xx!| ! s*!#33 88C r1ct|Sr^)rr s r/hflipr S>r1c|jdkDr|jddkDr t|dStj|dS)Nr r rrrr s r/ vflip_cv2rrr1c|ddddfSrrr s r/ vflip_numpyrs ttSy>r1ct|Sr^)rr s r/vfliprrr1c|jdd\}}|jdk(rdn|jd}|jdk(s|dkrtj||Sd}g}t d||D]{}t ||z|}|dddd||f} tj| |} | jdk(r&|jdk(rt j| d} |j| }t j|dS)a%Process images with more than 512 channels by splitting into chunks. OpenCV's flip function has a limitation where it can only handle images with up to 512 channels. This function works around that limitation by splitting the image into chunks of 512 channels, flipping each chunk separately, and then concatenating the results. Args: img: Input image with many channels flip_code: OpenCV flip code (0 for vertical, 1 for horizontal, -1 for both) Returns: Flipped image with all channels preserved Nr r4r rr) r"rhrVrrXrr$rappend concatenate) r6rheightwidthrY chunk_size result_chunksr[end_idxchunk flipped_chunks r/rrsIIbqMMFEA 1399Q>-a 00r1c0tdfd }|S)arDecorator to ensure float32 input/output for image processing functions. This decorator converts the input image to float32 before passing it to the wrapped function, and then converts the result back to the original dtype if it wasn't float32. Args: func (Callable[..., np.ndarray]): The image processing function to be wrapped. Returns: Callable[..., np.ndarray]: A wrapped function that handles float32 conversion. Example: @float32_io def some_image_function(img: np.ndarray) -> np.ndarray: # Function implementation return processed_img c|j}|tjk7r t|}|g|i|}|tjk7r t ||S|SN)r)r!r$rDrrr6argskwargs input_dtyperfuncs r/float32_wrapperz#float32_io..float32_wrapper*sSii "** $3-Cc+D+F+?Jbjj?Xz&{;d^ddr1r6 np.ndarrayr-rr.rreturnr3r)r0r1s` r/ float32_ior5s%& 4[ee r1c0tdfd }|S)ahDecorator to ensure uint8 input/output for image processing functions. This decorator converts the input image to uint8 before passing it to the wrapped function, and then converts the result back to the original dtype if it wasn't uint8. Args: func (Callable[..., np.ndarray]): The image processing function to be wrapped. Returns: Callable[..., np.ndarray]: A wrapped function that handles uint8 conversion. Example: @uint8_io def some_image_function(img: np.ndarray) -> np.ndarray: # Function implementation return processed_img c|j}|tjk7rt|tj}|g|i|}|tjk7r t |S|Sr+)r!r$rArrr,s r/ uint8_wrapperzuint8_io..uint8_wrapperIsWii "(( "Srxx8Cc+D+F+#."((#:xFFr1r2r)r0r8s` r/uint8_ior96s%& 4[GG r1) r)r3r*rUr+r3r,rUr4r3)r6r3r7r3r4r3)r6r3r7rUr4r3)r!ztype[np.number]r7float | np.ndarrayrH#Literal['add', 'multiply', 'power']r4r3)T)r6r3rJr3rboolr4r3) r6r3r7r:rHr;rr<r4r3)r6r3r7np.ndarray | floatrHLiteral['add', 'multiply']r4r3)r6r3r7rUrHr>r4r=)r6r3r7r3rHr>r4r3)r6r3r7r:rHr;r4r3)r6r3r7r=rr<r4r3)r6r3r7r=r4r3)r6r3r7r:r4r3)r6r3r7rUrr<r4r3) r6r3r7r3rYrTrr<r4r3)F)r6r3r7rrr<r4r3)r6r3r7r3rr<r4r3)r6r3rr:rr:r4r3)r6r3rrrrr4r3)r6r3rr:r4r3)r6r3rr:rr<r4r3)r6r3rrrr<r4r3) r)r3r*rUr+r3r,rUrr<r4r3)r6r3rrr7rr4r3) r6r3rrr7rrr<r4r3)r6r3rrr4r3r^)r6r3rI float | Noner4r3)r6r3rznp.dtyperIr?r4r3)r6r3r4r3)r6r3rrTr4r3)r0Callable[..., np.ndarray]r4r@)] __future__r functoolsrtypingrrrrVnumpyr$simsimdr& stringzillarNalbucore.decoratorsr r albucore.utilsr r r rrrrrrrrrrrFrcv2_operationsr0r8r<r?rKrQr\rbr_r`rmrrrtrwryr{r}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrr5r9rr1r/rJs{")) @    [["((K !ll377SWWM 2Cf< < <3< <&  \ \ \3\ \  \$     *    * $  * "O O O3O O6$$/ ' ' ' ' ' ' )((**1 + + + +  + + b  CC&TT(4/ c c"6  &  &i<<  1 1 1 1 1  1  11> > >; e e\( 3 30G0GfGG@'G'GT:< ,,"*F GG":B    '1T>r1