^jO$FdZddlmZddlZddlmZddlmZmZmZm Z m Z ddl Z ddl Z ddlmZmZmZddlmZmZddlmZerdd lmZgd Zed Ze d e j6 Ze dedef ZddZddZddZ ddZ!ddZ"ddZ#GddZ$ddZ%y)aModule containing utility functions for augmentation operations. This module provides a collection of helper functions and utilities used throughout the augmentation pipeline. It includes functions for image loading, type checking, error handling, mathematical operations, and decorators that add functionality to other functions in the codebase. These utilities help ensure consistent behavior and simplify common operations across different augmentation transforms. ) annotationsNwraps) TYPE_CHECKINGAnyCallableTypeVarcast)is_grayscale_imageis_multispectral_image is_rgb_image) Concatenate ParamSpec)angle_to_2pi_range)Path)angle_2pi_range non_rgb_errorread_bgr_imageread_grayscaleread_rgb_imagePT)boundF.c\tjt|tjS)zRead an image in BGR format from the specified path. Args: path (str | Path): Path to the image file. Returns: np.ndarray: Image in BGR format as a numpy array. )cv2imreadstr IMREAD_COLORpaths m/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/albumentations/augmentations/utils.pyrr,s ::c$i!1!1 22c`t|}tj|tjS)a%Read an image in RGB format from the specified path. This function reads an image in BGR format using OpenCV and then converts it to RGB format. Args: path (str | Path): Path to the image file. Returns: np.ndarray: Image in RGB format as a numpy array. )rrcvtColor COLOR_BGR2RGB)r!images r"rr9s% 4 E <<s00 11r#c\tjt|tjS)zRead a grayscale image from the specified path. Args: path (str | Path): Path to the image file. Returns: np.ndarray: Grayscale image as a numpy array. )rrrIMREAD_GRAYSCALEr s r"rrJs ::c$i!5!5 66r#c0tdfd }|S)u$Decorator to normalize angle values to the range [0, 2π). This decorator wraps a function that processes keypoints, ensuring that angle values (stored in the 4th column, index 3) are normalized to the range [0, 2π) after the wrapped function executes. Args: func (Callable): Function that processes keypoints and returns a numpy array. The function should take a keypoints array as its first parameter. Returns: Callable: Wrapped function that normalizes angles after processing keypoints. c|g|i|}t|dkDr+|jddkDrt|dddf|dddf<|S)Nr)lenshaper) keypointsargskwargsresultfuncs r"wrapped_functionz)angle_2pi_range..wrapped_functionisPi1$1&1 v;?v||A2-fQTl;F1a4L r#)r0 np.ndarrayr1zP.argsr2zP.kwargsreturnr6r)r4r5s` r"rrWs#$ 4[ r#ctt|s-d}t|r|dz }t|r|dz }t|y)aCheck if the input image is RGB and raise a ValueError if it's not. This function is used to ensure that certain transformations are only applied to RGB images. It provides helpful error messages for grayscale and multi-spectral images. Args: image (np.ndarray): The input image to check. Expected to be a numpy array representing an image. Raises: ValueError: If the input image is not an RGB image (i.e., does not have exactly 3 channels). The error message includes specific instructions for grayscale images and a note about incompatibility with multi-spectral images. Note: - RGB images are expected to have exactly 3 channels. - Grayscale images (1 channel) will trigger an error with conversion instructions. - Multi-spectral images (more than 3 channels) will trigger an error stating incompatibility. Example: >>> import numpy as np >>> rgb_image = np.random.randint(0, 256, (100, 100, 3), dtype=np.uint8) >>> non_rgb_error(rgb_image) # No error raised >>> >>> grayscale_image = np.random.randint(0, 256, (100, 100), dtype=np.uint8) >>> non_rgb_error(grayscale_image) # Raises ValueError with conversion instructions >>> >>> multispectral_image = np.random.randint(0, 256, (100, 100, 5), dtype=np.uint8) >>> non_rgb_error(multispectral_image) # Raises ValueError stating incompatibility z,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 imagesN)r r r ValueError)r'messages r"rrssH@  @ e $ u uG !% ( Y YG!! r#c tfd|Dstd|ddd|d|dkst|d|y ) aChecks if the given value is within the specified bounds Args: value (tuple[float, float]): The value to check and convert. Can be a single float or a tuple of floats. lower_bound (float): The lower bound for the range check. upper_bound (float): The upper bound for the range check. name (str | None): 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. c3>K|]}|cxkxrkncywN).0x lower_bound upper_bounds r" zcheck_range..s>1{a.;..>szAll values in z must be within [z, z] for tuple inputs.rr,z2 tuple values must be ordered as (min, max). Got: N)allr9)valuerArBnames `` r" check_rangerGsd >> >>$/@ RP[}\opqq 8uQx D8#UV[U\]^^ r#cFeZdZd d dZd dZd dZd dZd dZd dZd dZ y)PCANc`||dkr td||_d|_d|_d|_y)Nrz/Number of components must be greater than zero.)r9 n_componentsmean components_explained_variance_)selfrKs r"__init__z PCA.__init__s:  # (9NO O('+ .26: r#c,|jtjd}|j\}}|jt |||_t j|d|j\|_}}||_ |j|_ y)NFcopy)rL maxComponents) astypenpfloat64r/rKminr PCACompute2rLrMflattenrN)rOr@ n_samples n_features eigenvectors eigenvaluess r"fitzPCA.fits~ HHRZZeH , ! :    $ #Iz :D /2qt[_[l[l/m, <'#.#6#6#8 r#c|j td|jtjd}t j ||j|jS)NgThis PCA instance is not fitted yet. Call 'fit' with appropriate arguments before using this estimator.FrR)rMr9rUrVrWr PCAProjectrLrOr@s r" transformz PCA.transformsV    #U  HHRZZeH ,~~aD,<,<==r#cF|j||j|Sr=)r_rdrcs r" fit_transformzPCA.fit_transforms  ~~a  r#c|j tdtj||j|jS)Nra)rMr9rPCABackProjectrLrcs r"inverse_transformzPCA.inverse_transformsB    #U !!!TYY0@0@AAr#c|j tdtj|j}|j|z S)NzdThis PCA instance is not fitted yet. Call 'fit' with appropriate arguments before using this method.)rNr9rVsum)rOtotal_variances r"explained_variance_ratiozPCA.explained_variance_ratiosE  # # +v  8 89''.88r#cHtj|jSr=)rVcumsumrm)rOs r"#cumulative_explained_variance_ratioz'PCA.cumulative_explained_variance_ratiosyy66899r#r=)rKz int | Noner7None)r@r6r7rq)r@r6r7r6)r7r6) __name__ __module__ __qualname__rPr_rdrfrirmrpr>r#r"rIrIs&; 9>!B9:r#rIcdfd }|S)Nc\tjdfd }td|S)Nct|dkDr|d}n|vr|}ntdt|dk(r|S|i|S)NrzMissing required argument: )r.r9)r1r2arrayr4 param_names r"wrapperz6handle_empty_array..decorator..wrappers^4y1}Qv%z* #>zl!KLL5zQ (( (r#r)r1rr2rr7r) functoolsrr )r4rzrys` r" decoratorz%handle_empty_array..decorators-   )  )C!!r#)r4rr7rr>)ryr|s` r"handle_empty_arrayr}s"$ r#)r!z str | Pathr7r6)r40Callable[Concatenate[np.ndarray, P], np.ndarray]r7r~)r'r6r7rq) rEztuple[float, float]rAfloatrBrrFz str | Noner7rq)ryrr7zCallable[[F], F])&__doc__ __future__rr{rtypingrrrr r rnumpyrValbucore.utilsr r r typing_extensionsrr#albumentations.core.keypoints_utilsrpathlibr__all__rndarrayrrrrrrrrGrIr}r>r#r"rs#>>  5B cN Crzz" CxS)* 32" 7 :58'"T_&3:3:lr#