U h(! @sHddlmZddlmZddlZddlZddlmZm Z ddl m Z ddl mZddlmZmZddlmZd d d d d dddgZdddddddd Zddddddd Zedddddd Zdddddd Zddddddddd Ze dd d d!dddd"dd# d$dZdd%d%dddd&d'dZed,dd%d%ddd"dd*d+dZdS)-) annotations)SequenceN)maybe_process_in_chunkspreserve_channel_dim) functional)handle_empty_array)denormalize_bboxesnormalize_bboxes) ColorTypeget_crop_coordscrop_bboxes_by_coordscrop_keypoints_by_coordsget_center_crop_coordscrop crop_and_padcrop_and_pad_bboxescrop_and_pad_keypointsztuple[int, int]floatztuple[int, int, int, int]) image_shape crop_shapeh_startw_startreturnc Csd|dd\}}|dd\}}t||d|}||} t||d|} | |} | || | fS)N)int) rrrrheightwidth crop_height crop_widthy_miny_maxx_minx_maxr$Q/tmp/pip-unpacked-wheel-e8onvpoz/albumentations/augmentations/crops/functional.pyr s z np.ndarray)bboxes crop_coordsrrc Cs|js |St|tj|}|dd\}}|ddddgf|8<|ddddgf|8<|d|d}|d|d}||f}t||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 should be normalized (in the range [0, 1]). 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). Returns: np.ndarray: Array of cropped bounding boxes, normalized to the new crop size. 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)sizercopyZastypenpZfloat32r ) r&r'rZcropped_bboxesr"r rrrr$r$r%r 0s) keypointsr'rcCsL|dd\}}|}|dddf|8<|dddf|8<|S)aCrop keypoints using the provided coordinates of bottom-left and top-right corners in pixels. Args: keypoints (np.ndarray): An array of keypoints with shape (N, 4+) where each row is (x, y, angle, scale, ...). crop_coords (tuple): Crop box coords (x1, y1, x2, y2). Returns: np.ndarray: An array of cropped keypoints with the same shape as the input. Nrrr)r*)r,r'x1y1Zcropped_keypointsr$r$r%r \s )rrrc CsT|dd\}}|dd\}}||d}||}||d}||} ||| |fS)Nrr$) rrrrrrr r!r"r#r$r$r%rss  r)imgr"r r#r!rcCs|jdd\}}||ks"||krDtd|d|d|d|d |dksd||ksd|dksd||krtd|d|d|d|d |d |d |||||fS) NrzDWe should have x_min < x_max and y_min < y_max. But we got (x_min = z , y_min = z , x_max = z , y_max = )rzUValues for crop should be non negative and equal or smaller than image sizes(x_min = z , height = z , width = )shape ValueError)r/r"r r#r!rrr$r$r%r~s (zSequence[int] | NonezColorType | Nonebool) r/ crop_params pad_params pad_valuer interpolationpad_mode keep_sizerc Cs|dk r(tdd|Dr(t|f|}|dk rjtdd|Drjtj||d|d|d|d||d}|r|dd\}} ttj| |f|d } | |S|S) Ncss|]}|dkVqdSrNr$.0ir$r$r% szcrop_and_pad..css|]}|dkVqdSr:r$r;r$r$r%r>srrrr()Z border_modevalue)Zdsizer7)anyr fgeometricZpad_with_paramsrcv2resize) r/r4r5r6rr7r8r9rowscolsZ resize_fnr$r$r%rs"  z tuple[int, int, int, int] | None)r&r4r5r result_shaperc Cst|dkr|St||}|dk rj|dd\}}|ddddgf|8<|ddddgf|8<|dk r|\}} } } |ddddgf| 7<|ddddgf|7<t||S)Nrrrr()lenrr ) r&r4r5rrFZdenormalized_bboxesZcrop_xZcrop_ytop_leftr$r$r%rs   rrF)r,r4r5rrFr9rcCs|}|dk rP|dd\}}|dddf|8<|dddf|8<|dk r|\} } } } |dddf| 7<|dddf| 7<|dd\} } |dd\}}|r|| ks|| kr| |}| |}t|||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*rAZkeypoints_scale)r,r4r5rrFr9Ztransformed_keypointsZcrop_x1Zcrop_y1rHrIrJrDrEZ result_rowsZ result_colsZscale_xZscale_yr$r$r%rs  )NNrKrKF) __future__rtypingrrBZnumpyr+ZalbucorerrZ&albumentations.augmentations.geometricrrAZ"albumentations.augmentations.utilsrZalbumentations.core.bbox_utilsrr Zalbumentations.core.typesr __all__r r r rrrrrr$r$r$r%sB      ,