biyddlmZddlZddlmZddlmZmZmZm Z ejdk\r ddlm Z m Z nddl m Z m Z ddlZddlZdZdZdZd ZdZd Zed Ze d Zej2d d ej2ddej2ddej2ddej2ddej2ddej4d ej6dej8dej:dej<dej>dej@di Z!ej4ejDej6ejFej<ejHej>ejJej@ejLej2d ejDej2dejFej2dejHej2dejJej2dejLi Z' d"dZ(d#d$dZ)d%dZ*d&dZ+d'dZ,d(dZ-d'dZ.d'dZ/d)d Z0e ejbe2e3fZ4d*d!Z5y)+) annotationsNwraps)AnyCallableLiteralUnion) ) Concatenate ParamSpecr )imageimage_per_channelmin_maxmin_max_per_channelPuint8uint16iuint32lfloat16g?float32float64iint32c8tdfd }|S)aWrap OpenCV function to enable processing images with more than 4 channels. Limitations: This wrapper requires image to be the first argument and rest must be sent via named arguments. Args: process_fn: Transform function (e.g cv2.resize). args: Additional positional arguments. kwargs: Additional keyword arguments. Returns: np.ndarray: Transformed image. cg |} |z}t|}|tkDrg}td|dD]}||z tk(r[tdD]L}|dddd||z||zdzf}  | g|i|} t j | d} |j | Nj|dddd||dzf}  | g|i|} |j | t j|S |g|i|S)Nrrr)get_num_channelsMAX_OPENCV_WORKING_CHANNELSrangeTWOnp expand_dimsappenddstack) img process_argsprocess_kwargsall_args all_kwargs num_channelschunksindexichunkargskwargs process_fns I/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/albucore/utils.py __process_fnz-maybe_process_in_chunks..__process_fnLs"*T)L)%+n%< ', 5 5Fq,2 )%'3."1X- #Aq%!)eai!m*C$C D *5 J8 Jz J "ub 9 e, -  1eeai&7 78E&uFxF:FEMM%( )99V$ $#77J77)r) np.ndarrayr*P.argsr+P.kwargsreturnr9r)r5r3r4r7s``` r6maybe_process_in_chunksr=9s$& :880 r8ct|}|rtj|d||Stj|d|j|dS)Nr)outF)copy)MAX_VALUES_BY_DTYPEr%clipastype)r)dtypeinplace max_values r6rBrBhsG#E*IwwsAyc22 7739 % , ,U , ??r8c0tdfd }|S)Nc|j}|g|i|}|jtjk(r|St||SN)rDr%rrB)r)r3r4rDresultfuncs r6wrapped_functionz!clipped..wrapped_functionpsB c+D+F+ <<288 #MFE""r8)r)r9r3r:r4r;r<r9r)rKrLs` r6clippedrMos! 4[## r8cJ|jtk(r|jdSdS)Nrr)ndimNUM_MULTI_CHANNEL_DIMENSIONSshapers r6r!r!}s!"ZZ+GG5;;q>NQNr8ct|dk(S)Nrr!rRs r6is_grayscale_imagerUs E "a ''r8c`t|tjr |j}t|SrI) isinstancer%ndarrayrDNPDTYPE_TO_OPENCV_DTYPE)values r6get_opencv_dtype_from_numpyr[s$%$  "5 ))r8c&t|tk(SrI)r!NUM_RGB_CHANNELSrRs r6 is_rgb_imager^s E "&6 66r8c t|}|dvS)N>rr rT)rr.s r6is_multispectral_imager`s#E*L v %%r8ct|tttjtj fr1t|ttfr t|S|j St|tjrd|jdk(r|j S|jdkDr|St|dk(s|dk(st||krt|dS|d|Stdt|)aConvert a value to a float or numpy array based on its shape and number of channels. Args: value: Input value to convert (numpy array, float, or int) num_channels: Number of channels in the target image Returns: float: If value is a scalar or 1D array that should be converted to scalar np.ndarray: If value is a multi-dimensional array or channel vector Raises: TypeError: If value is of unsupported type rrNzUnsupported value type: ) rWfloatintr%rritemrXrOlen TypeErrortype)rZr.s r6 convert_valuerhs%%bjj"**=>)%%>uU|PEJJLP%$ ::?::<  ::>L u:?la/3u: 3Lq? "]l## .tE{m< ==r8cF|tvrd|d}t|t|S)Nz(Can't infer the maximum value for dtype zS. You need to specify the maximum value manually by passing the max_value argument.)rA RuntimeError)rDmsgs r6 get_max_valuerls; ''6ug>` ` 3 u %%r8)r50Callable[Concatenate[np.ndarray, P], np.ndarray]r3r:r4r;r<z"Callable[[np.ndarray], np.ndarray])F)r)r9rDrrEboolr<r9)rKrmr<rm)rr9r<rc)rr9r<rn)rZz$np.ndarray | int | np.dtype | objectr<rc)rZznp.ndarray | floatr.rcr<zfloat | np.ndarray)rDznp.dtyper<rb)6 __future__rsys functoolsrtypingrrrr version_infor r typing_extensionscv2numpyr%r]MONO_CHANNEL_DIMENSIONSrPFOURr$r"NormalizationTyperrDrrrrrrrrACV_8UCV_16UCV_32FCV_64FCV_32SrYr=rBrMr!rUr[r^r`rhrXrbrc ValueTyperlr8r6rs1" 00w--8  Z[ cN BHHWs BHHX BHHX  BHHY BHHY BHHYHHcIIuIIzJJJJJJHHj"HHciiIIszzJJ JJ HHcjj BHHWsyy BHHX  BHHY BHHY BHHWszz ,@, ,,( ,^@ O(* 7& ">J "**eS( ) &r8