a aDfY@sZddlmZddlmZddlmZmZmZmZm Z ddl Z ddl Z ddl mZmZddlmZddlmZmZmZmZgdZed Ze d d e d d e dde dde jd e jd e jde jdiZe je je je j e j!e j"e je j#e j$e j%e d e je d e j e de j"e de j#e de j%i Z&dZ'dZ(ee jZ)e e*efe j+dddZ,e e*efe j+dddZ-e e*efe j+dddZ.eee j+efe j+feee j+efe j+fdddZ/e j+e je0e j+d d!d"Z1e e j+e2e je3fe2d#d$d%Z4eeeefefeeeefefdd&d'Z5eee j+efe j+feee j+efe j+fdd(d)Z6eee j+efe j+feee j+efe j+fdd*d+Z7e j+e2d,d-d.Z8e j+e9d,d/d0Z:e j+e9d,d1d2Z;e j+e9d,d3d4Zee0e0fe0e0ee*dd:d;d<Z?dS)=)wraps)Path)AnyCallableOptionalTupleUnionN) Concatenate ParamSpecangle_to_2pi_range)MONO_CHANNEL_DIMENSIONSNUM_MULTI_CHANNEL_DIMENSIONSNUM_RGB_CHANNELSKeypointInternalType)read_bgr_imageread_rgb_imageread_grayscaleMAX_VALUES_BY_DTYPENPDTYPE_TO_OPENCV_DTYPEclippedget_opencv_dtype_from_numpyangle_2pi_rangeclippreserve_channel_dimensure_contiguous is_rgb_imageis_grayscale_imageis_multispectral_imageget_num_channelsnon_rgb_warning_maybe_process_in_chunksPuint8uint16iuint32lfloat32?int32float64)pathreturncCstt|tjSN)cv2imreadstrZ IMREAD_COLORr-r4o/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/albumentations/augmentations/utils.pyrDsrcCst|}t|tjSr/)rr0ZcvtColorZ COLOR_BGR2RGB)r-imager4r4r5rHsrcCstt|tjSr/)r0r1r2ZIMREAD_GRAYSCALEr3r4r4r5rMsr)funcr.cs,ttjtjtjtjdfdd }|S)Nimgargskwargsr.cs0|j}t|d}t|g|Ri|||S)Nr()dtypergetr)r9r:r;r<maxvalr7r4r5wrapped_functionRs z!clipped..wrapped_functionrnpndarrayr"r:r;r7r@r4r?r5rQs"r)r9r<r>r.cCst|d||S)Nr)rBrZastype)r9r<r>r4r4r5r[sr)valuer.cCst|tjr|j}t|S)zReturn a corresponding OpenCV dtype for a numpy's dtype :param value: Input dtype of numpy array :return: Corresponding dtype for OpenCV ) isinstancerBrCr<r)rEr4r4r5r_s rcs(tttjtjtdfdd }|S)N)keypointr:r;r.cs6|g|Ri|dd\}}}}||t||fS)Nr,r )rGr:r;xyasr?r4r5r@ls&z)angle_2pi_range..wrapped_function)rrr"r:r;rDr4r?r5risrcs,ttjtjtjtjdfdd }|S)zPreserve dummy channel dim.r8cs|j}|g|Ri|}t|tkrP|ddkrPt|jtkrPtj|ddSt|tkrt|jtkr|dddddfS|S)N)Zaxisr)shapelenrr rB expand_dims)r9r:r;rNresultr?r4r5r@ys   z.preserve_channel_dim..wrapped_functionrArDr4r?r5rts"rcs,ttjtjtjtjdfdd }|S)z$Ensure that input img is contiguous.r8cs&tj|dgd}|g|Ri|S)NZ C_CONTIGUOUS) requirements)rBrequire)r9r:r;r?r4r5r@sz+ensure_contiguous..wrapped_functionrArDr4r?r5rs"r)r6r.cCst|jtkr|jdSdS)Nr+rM)rOrNrr6r4r4r5rsrcCs t|tkSr/)rrrTr4r4r5rsrcCs t|dkS)NrMrrTr4r4r5rsrcCst|}|dvS)N>rMrU)r6 num_channelsr4r4r5rsrcCs8t|s4d}t|r|d7}t|r,|d7}t|dS)Nz,This transformation expects 3-channel imagesz[ You can convert your grayscale image to RGB using cv2.cvtColor(image, cv2.COLOR_GRAY2RGB))z? This transformation cannot be applied to multi-spectral images)rrr ValueError)r6messager4r4r5r sr ) process_fnr;r.c s&ttjtjdfdd }|S)anWrap 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). kwargs: Additional parameters. Returns: numpy.ndarray: Transformed image. )r9r.cst|}|tkrg}td|dD]}||tkrtdD]P}|dddd||||df}|fi}t|d}||q8q |dddd||df}|fi}||q t|S|fiS)Nrr,r+rMrL)rFOURrangeTWOrBrPappendZdstack)r9rWchunksindexichunkr;rZr4r5 __process_fns  &   z._maybe_process_in_chunks..__process_fn)rrBrC)rZr;rdr4rcr5r!sr!)rE lower_bound upper_boundnamer.csZtfdd|Ds4td|ddd|d|dksVt|d |d S) aChecks if the given value is within the specified bounds Args: value: The value to check and convert. Can be a single float or a tuple of floats. lower_bound: The lower bound for the range check. upper_bound: The upper bound for the range check. name: The name of the parameter being checked. Used for error messages. Raises: ValueError: If the value is outside the bounds or if the tuple values are not ordered correctly. c3s&|]}|koknVqdSr/r4).0rHrerfr4r5 zcheck_range..zAll values in z must be within [z, z] for tuple inputs.rrMz2 tuple values must be ordered as (min, max). Got: N)allrX)rErerfrgr4rir5 check_ranges rm)@ functoolsrpathlibrtypingrrrrrr0numpyrBZtyping_extensionsr r Z#albumentations.core.keypoints_utilsr Zalbumentations.core.typesr rrr__all__r"r<r#r%r&r'rZCV_8UZCV_16Ur)ZCV_32SZCV_32Fr*ZCV_64Frr]r[Z BIG_INTEGERr2rCrrrrfloatrintobjectrrrrrboolrrrr r!rmr4r4r4r5sr               8       *