U h2@sLddlmZddlZddlmZddlmZmZmZmZm Z ddl Z ddl Z ddl mZmZmZddlmZmZddlmZerddlmZd d d d d gZedZede jdZededefdZddddd Zddddd Zddddd Zddddd Z ddddd Z!dd d d!dd"d#d$Z"Gd%d&d&Z#dddd'd(Z$dS))) annotationsNwraps) TYPE_CHECKINGAnyCallableTypeVarcast)is_grayscale_imageis_multispectral_image is_rgb_image) Concatenate ParamSpec)angle_to_2pi_range)Pathread_bgr_imageread_rgb_imageread_grayscaleangle_2pi_range non_rgb_errorPT)boundF.z str | Path np.ndarray)pathreturncCstt|tjSN)cv2imreadstrZ IMREAD_COLORrr"F/tmp/pip-unpacked-wheel-e8onvpoz/albumentations/augmentations/utils.pyr#scCst|}t|tjSr)rrZcvtColorZ COLOR_BGR2RGB)rimager"r"r#r'scCstt|tjSr)rrr ZIMREAD_GRAYSCALEr!r"r"r#r,sz0Callable[Concatenate[np.ndarray, P], np.ndarray])funcrcs$tdddddfdd }|S)NrzP.argszP.kwargs) keypointsargskwargsrcsN|f||}t|dkrJ|jddkrJt|dddf|dddf<|S)Nr)lenshaper)r&r'r(resultr%r"r#wrapped_function3s z)angle_2pi_range..wrapped_functionr)r%r/r"r.r#r0sNone)r$rcCs8t|s4d}t|r|d7}t|r,|d7}t|dS)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$messager"r"r#r=sztuple[float, float]floatz str | None)value lower_bound upper_boundnamercsZtfdd|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&|]}|koknVqdSrr").0xr5r6r"r# rszcheck_range..zAll values in z must be within [z, z] for tuple inputs.rr)z2 tuple values must be ordered as (min, max). Got: N)allr1)r4r5r6r7r"r:r# check_rangefs r=c@szeZdZddddddZdddd d Zdddd d Zdddd dZdddddZddddZddddZ dS)PCANz int | Noner0) n_componentsrcCs4|dk r|dkrtd||_d|_d|_d|_dS)Nrz/Number of components must be greater than zero.)r1r?mean components_explained_variance_)selfr?r"r"r#__init__ys z PCA.__init__r)r9rcCsZ|tj}|j\}}|jdkr,t|||_tj|d|jd\|_}}||_ | |_ dS)N)r@Z maxComponents) astypenpfloat64r,r?minrZ PCACompute2r@rAflattenrB)rCr9Z n_samplesZ n_featuresZ eigenvectorsZ eigenvaluesr"r"r#fits    zPCA.fitcCs0|jdkrtd|tj}t||j|jSNzgThis PCA instance is not fitted yet. Call 'fit' with appropriate arguments before using this estimator.)rAr1rErFrGrZ PCAProjectr@rCr9r"r"r# transforms   z PCA.transformcCs||||Sr)rJrMrLr"r"r# fit_transforms zPCA.fit_transformcCs$|jdkrtdt||j|jSrK)rAr1rZPCABackProjectr@rLr"r"r#inverse_transforms  zPCA.inverse_transform)rcCs(|jdkrtdt|j}|j|S)NzdThis PCA instance is not fitted yet. Call 'fit' with appropriate arguments before using this method.)rBr1rFsum)rCZtotal_variancer"r"r#explained_variance_ratios   zPCA.explained_variance_ratiocCst|Sr)rFZcumsumrQ)rCr"r"r##cumulative_explained_variance_ratiosz'PCA.cumulative_explained_variance_ratio)N) __name__ __module__ __qualname__rDrJrMrNrOrQrRr"r"r"r#r>xs   r>cs,tdddddfdd }tt|S)Nrr)arrayr'r(rcs t|dkr|S|f||S)Nr)r+)rVr'r(r.r"r#wrappers z#handle_empty_array..wrapper) functoolsrr r)r%rWr"r.r#handle_empty_arraysrY)% __future__rrXrtypingrrrrr rZnumpyrFZalbucore.utilsr r r Ztyping_extensionsr rZ#albumentations.core.keypoints_utilsrpathlibr__all__rZndarrayrrrrrrrr=r>rYr"r"r"r#s6     )7