^jBdZddlmZddlmZddlmZddlZddlZ ddl m Z m Z ddl mZddlmZdd lmZmZgd Z dd Z d dd Zed  ddZddZddZe ddZ ddZed  d d dZ d!dZ d"dZ d# d$dZ y)%aFunctional implementations of image cropping operations. This module provides utility functions for performing various cropping operations on images, bounding boxes, and keypoints. It includes functions to calculate crop coordinates, crop images, and handle the corresponding transformations for bounding boxes and keypoints to maintain consistency between different data types during cropping operations. ) annotations)Sequence)AnyN)maybe_process_in_chunkspreserve_channel_dim) functional)handle_empty_array)denormalize_bboxesnormalize_bboxes) crop crop_and_padcrop_and_pad_bboxescrop_and_pad_keypointscrop_bboxes_by_coordscrop_keypoints_by_coordsget_center_crop_coordsget_crop_coordspad_along_axesvolume_crop_yxvolumes_crop_yxc|dd\}}t|d|}t|d|}t||z dz|z}||z} t||z dz|z} | |z} | || | fS)aSGet crop coordinates. This function gets the crop coordinates. Args: image_shape (tuple[int, int]): Original image shape. crop_shape (tuple[int, int]): Crop shape. h_start (float): Start height. w_start (float): Start width. Returns: tuple[int, int, int, int]: Crop coordinates. Nr)minint) image_shape crop_shapeh_startw_startheightwidth crop_height crop_widthy_miny_maxx_minx_maxs x/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/albumentations/augmentations/crops/functional.pyrr%s2 OMFEjmV,KZ]E*J +%)W4 5E K E #a'72 3E J E % %%c|js|S|r8t|jjtj |}n-|jjtj }|dd\}}|ddddgfxx|zcc<|ddddgfxx|zcc<|d|dz }|d|dz }||f} |r t || S|S)aCrop bounding boxes based on given crop coordinates. This function adjusts bounding boxes to fit within a cropped image. Args: bboxes (np.ndarray): Array of bounding boxes with shape (N, 4+) where each row is [x_min, y_min, x_max, y_max, ...]. The bounding box coordinates can be either normalized (in [0, 1]) if normalized_input=True or absolute pixel values if normalized_input=False. crop_coords (tuple[int, int, int, int]): Crop coordinates (x_min, y_min, x_max, y_max) in absolute pixel values. image_shape (tuple[int, int]): Original image shape (height, width). normalized_input (bool): Whether input boxes are in normalized coordinates. If True, assumes input is normalized [0,1] and returns normalized coordinates. If False, assumes input is in absolute pixels and returns absolute coordinates. Default: True for backward compatibility. Returns: np.ndarray: Array of cropped bounding boxes. Coordinates will be in the same format as input (normalized if normalized_input=True, absolute pixels if normalized_input=False). Note: Bounding boxes that fall completely outside the crop area will be removed. Bounding boxes that partially overlap with the crop area will be adjusted to fit within it. Nrrr)sizer copyastypenpfloat32r ) bboxes crop_coordsrnormalized_inputcropped_bboxesr&r$r"r#rs r(rrKs@ ;; +FKKM,@,@,LkZ--bjj9r?LE51q!f9&1q!f9&a.;q>1KQ+a.0Jz*Jzcrop_and_pad..s&C!qAv&Cc3&K|] }|dk7 ywrDr;rEs r(rHzcrop_and_pad..s%Aa1f%ArIrrrr+) border_modevalue)dsize interpolation)anyr fgeometricpad_with_paramsrcv2resize) rA crop_params pad_params pad_valuerrNpad_mode keep_sizerowscols resize_fns r(r r s:3&C{&C#C3%%#%Aj%A"A((  qM qM qM qM   !_ d+CJJtTlZgh ~ Jr)ct|dk(r|St||}|.|dd\}}|ddddgfxx|zcc<|ddddgfxx|zcc<|-|\}} } } |ddddgfxx| z cc<|ddddgfxx|z cc<t||S)aCrop and pad bounding boxes. This function crops and pads bounding boxes. Args: bboxes (np.ndarray): Array of bounding boxes. crop_params (tuple[int, int, int, int] | None): Crop parameters. pad_params (tuple[int, int, int, int] | None): Pad parameters. image_shape (tuple[int, int]): Original image shape. result_shape (tuple[int, int]): Result image shape. Returns: np.ndarray: Array of cropped and padded bounding boxes. rNrrr+)lenr r ) r1rTrUr result_shapedenormalized_bboxescrop_xcrop_ytop_lefts r(rrs, 6{a -V[A$RaA1vI&&0&A1vI&&0&$QaA1vI&$.&A1vI&#-& / >>r)cN|j}|*|dd\}}|dddfxx|zcc<|dddfxx|zcc<|)|\} } } } |dddfxx| z cc<|dddfxx| z cc<|dd\} } |dd\}}|r+|| k7s|| k7r!| |z }| |z }tj|||S|S)aCrop and pad multiple keypoints simultaneously. Args: keypoints (np.ndarray): Array of keypoints with shape (N, 4+) where each row is (x, y, angle, scale, ...). crop_params (Sequence[int], optional): Crop parameters [crop_x1, crop_y1, ...]. pad_params (Sequence[int], optional): Pad parameters [top, bottom, left, right]. image_shape (Tuple[int, int]): Original image shape (rows, cols). result_shape (Tuple[int, int]): Result image shape (rows, cols). keep_size (bool): Whether to keep the original size. Returns: np.ndarray: Array of transformed keypoints with the same shape as input. Nrrr)r-rPkeypoints_scale)r5rTrUrr^rXtransformed_keypointscrop_x1crop_y1rbrcrdrYrZ result_rows result_colsscale_xscale_ys r(rr1s.&NN,&r?ad#w.#ad#w.#$Qaad#t+#ad#s*#RaJD$+BQ/KkT)[D-@$$))*?'RR  r)c|jdd\}}}||ks||krtd|d|d|d|d |dks|dks ||kDs||kDr'td |d|d|d|d |jdd |dd||||fS) aCrop a single volume along Y (height) and X (width) axes only. Args: volume (np.ndarray): Input volume with shape (D, H, W) or (D, H, W, C). x_min (int): Minimum width coordinate. y_min (int): Minimum height coordinate. x_max (int): Maximum width coordinate. y_max (int): Maximum height coordinate. Returns: np.ndarray: Cropped volume (D, H_new, W_new, [C]). Raises: ValueError: If crop coordinates are invalid. Nr+5Crop coordinates must satisfy min < max. Got: (x_min=, y_min=, x_max=, y_max=r=rECrop coordinates must be within image dimensions (H, W). Got: (x_min=) for volume shape r>)volumer&r$r'r%rcr r!s r(rr_s.||BQ'Avu ~% WHUG8E7(5' L   qyEAI%&. WHUG8E7(5'J & Ra 01 3   !U5[%+- ..r)cNd|jcxkrdksntd|j|jdd\}}}||ks||krtd|d|d|d|d |d ks|d ks ||kDs||kDrtd |d|d|d|d |||f |d d d d ||||fS)a'Crop a batch of volumes along Y (height) and X (width) axes only. Args: volumes (np.ndarray): Input batch of volumes with shape (B, D, H, W) or (B, D, H, W, C). x_min (int): Minimum width coordinate. y_min (int): Minimum height coordinate. x_max (int): Maximum width coordinate. y_max (int): Maximum height coordinate. Returns: np.ndarray: Cropped batch of volumes (B, D, H_new, W_new, [C]). Raises: ValueError: If crop coordinates are invalid or volumes shape is incorrect. z1Input volumes should have 4 or 5 dimensions, got rrorprqrrr=rrsrtN)ndimr@r?)volumesr&r$r'r%depthr r!s r(rrs.  ! !LW\\N[\\"==1-E65 ~% WHUG8E7(5' L   qyEAI%&. WHUG8E7(5'J !& 67 9   1aueEk1 22r)c |j} d|cxkr| krnn d|cxkr| ksntd|d|d| d||k(rtd|d|dtjd tjd tj d tj d tjd i} || vrtd|| |} dg| z} ||f| |<||f| |<d| i} | d k(r|| d<tj|| fi| S)aPad an array along specified height (H) and width (W) axes using np.pad. Args: arr (np.ndarray): Input array. pad_top (int): Padding added to the top (start of H axis). pad_bottom (int): Padding added to the bottom (end of H axis). pad_left (int): Padding added to the left (start of W axis). pad_right (int): Padding added to the right (end of W axis). h_axis (int): Index of the height axis (Y). w_axis (int): Index of the width axis (X). border_mode (int): OpenCV border mode. pad_value (float | Sequence[float]): Value for constant padding. Returns: np.ndarray: Padded array. Raises: ValueError: If border_mode is unsupported or axis indices are out of bounds. rz Axis indices z, z" are out of bounds for array with z dimensions.z Height axis z and width axis z cannot be the same.constantedgereflect symmetricwrapzUnsupported border_mode: rrmodeconstant_values) ryr@rRBORDER_CONSTANTBORDER_REPLICATEBORDER_REFLECTBORDER_REFLECT_101 BORDER_WRAPr/pad)arrpad_top pad_bottompad_left pad_righth_axisw_axisrKrVrymode_mapnp_mode pad_widthkwargss r(rrs1> 88D  $ 1#5#5=6(:\]a\bbnopp <x/?xG[\]] Z f I    H("4[MBCC{#G4I *-If!9-If%g.F*$- ! 66#y +F ++r)) rtuple[int, int]rrrfloatrrreturntuple[int, int, int, int])T) r1 np.ndarrayr2rrrr3boolrr)r5rr2rrr)rrrrrr) rArr&rr$rr'rr%rrr)rArrT tuple[int, int, int, int] | NonerUrrVz tuple[float, ...] | float | NonerrrNrrWrrXrrr) r1rrTrrUrrrr^rrr)NNrrF)r5rrTrrUrrrr^rrXrrr) rurr&rr$rr'rr%rrr) rzrr&rr$rr'rr%rrr)r)rrrrrrrrrrrrrrrKrrVzfloat | Sequence[float]rr)!__doc__ __future__rcollections.abcrtypingrrRnumpyr/albucorerr&albumentations.augmentations.geometricrrP"albumentations.augmentations.utilsr albumentations.core.bbox_utilsr r __all__rrrrr r rrrrrr;r)r(rs#$ BKAO #& #&#&#& #&  #&T" 5` 5`*5`!5` 5`  5`pK *!.&.)B. .1.1.0 . ! .  .....b)? )?1)?1)?! )? " )?  )?XK 5937#)$* *!*!1*!1*!! *! " *!  *!*!!*!Z&/ &/ &/ &/  &/  &/  &/R)3 )3 )3 )3  )3  )3  )3j*+:, :, :,:, :,  :,  :, :,:,':,:,r)