hJSrSSKJr SSKrSSKJr SSKJrJr SSK r SSK J r SSK JrJrJr /S Qr1S krSS jr"S S \5r"SS\5rSSjrSSjrSSSjjrSSSjjrg)aModule for handling keypoint operations during augmentation. This module provides utilities for working with keypoints in various formats during the augmentation process. It includes functions for converting between coordinate systems, filtering keypoints based on visibility, validating keypoint data, and applying transformations to keypoints. The module supports different keypoint formats including xy, yx, and those with additional angle or size information. ) annotationsN)Sequence)AnyLiteral)'NUM_KEYPOINTS_COLUMNS_IN_ALBUMENTATIONS) DataProcessorParams ShapeType)KeypointParamsKeypointsProcessorangle_to_2pi_rangecheck_keypoints%convert_keypoints_from_albumentations#convert_keypoints_to_albumentationsfilter_keypoints>xyyxxyaxysxyzxyasxysacR[R"US[R-5$)uJConvert angles to the range [0, 2π). This function takes an array of angles and ensures they are all within the range of 0 to 2π (exclusive) by applying modulo 2π. Args: angles (np.ndarray): Array of angle values in radians. Returns: np.ndarray: Array of the same shape as input with angles normalized to [0, 2π). )npmodpi)angless ]/var/www/fran/franai/venv/lib/python3.13/site-packages/albumentations/core/keypoints_utils.pyrr#s 66&!bee) $$c^\rSrSrSrS S U4SjjjrS U4Sjjr\S Sj5r\S Sj5r S Sjr Sr U=r $)r 3aParameters of keypoints Args: format (str): format of keypoints. Should be 'xy', 'yx', 'xya', 'xys', 'xyas', 'xysa', 'xyz'. x - X coordinate, y - Y coordinate z - Z coordinate (for 3D keypoints) s - Keypoint scale a - Keypoint orientation in radians or degrees (depending on KeypointParams.angle_in_degrees) label_fields (list[str]): list of fields that are joined with keypoints, e.g labels. Should be same type as keypoints. remove_invisible (bool): to remove invisible points after transform or not angle_in_degrees (bool): angle in degrees or radians in 'xya', 'xyas', 'xysa' keypoints check_each_transform (bool): if `True`, then keypoints will be checked after each dual transform. Default: `True` Note: The internal Albumentations format is [x, y, z, angle, scale]. For 2D formats (xy, yx, xya, xys, xyas, xysa), z coordinate is set to 0. For formats without angle or scale, these values are set to 0. cH>[TU]X5 X0lX@lXPlgN)super__init__remove_invisibleangle_in_degreescheck_each_transform)selfformat label_fieldsr(r)r* __class__s r r'KeypointParams.__init__Ps$ . 0 0$8!r!c>[TU]5nURURURUR S.5 U$)zGet the private dictionary representation of keypoint parameters. Returns: dict[str, Any]: Dictionary containing the keypoint parameters. )r(r)r*)r&to_dict_privateupdater(r)r*)r+datar.s r r1KeypointParams.to_dict_private]sEw&( $($9$9$($9$9(,(A(A   r!cg)zCheck if the keypoint parameters are serializable. Returns: bool: Always returns True as KeypointParams is serializable. Tclss r is_serializableKeypointParams.is_serializablensr!cg)zQGet the full name of the class. Returns: str: The string "KeypointParams". r r6r7s r get_class_fullname!KeypointParams.get_class_fullnamexs r!c SURSURSURSURSURS3 $)NzKeypointParams(format=z, label_fields=z, remove_invisible=z, angle_in_degrees=z, check_each_transform=))r,r-r(r)r*r+s r __repr__KeypointParams.__repr__sY$T[[MARAR@ST!!%!6!6 77J4K`K`Jab%%)%>%>$?q B r!)r)r*r()NTTT) r,strr-zSequence[str] | Noner(boolr)rDr*rD)returndict[str, Any])rErDrErC) __name__ __module__ __qualname____firstlineno____doc__r'r1 classmethodr9r<rA__static_attributes__ __classcell__r.s@r r r 3s>.2!%!%%) 9 9+ 9 9  9 # 9 9"    r!r c^\rSrSrSrS S U4Sjjjr\S Sj5rSSjrSSjr SSjr SSjr SS jr S r U=r$)r aProcessor for keypoint data transformation. This class handles the conversion, validation, and filtering of keypoints during transformations. It ensures keypoints are correctly formatted and processed according to the specified keypoint parameters. Args: params (KeypointParams): Parameters for keypoint processing. additional_targets (dict[str, str] | None): Dictionary mapping additional target names to their types. c$>[TU]X5 gr%)r&r')r+paramsadditional_targetsr.s r r'KeypointsProcessor.__init__s 4r!cg)zTGet the default name for keypoint data. Returns: str: The string "keypoints". keypointsr6r@s r default_data_name$KeypointsProcessor.default_data_namesr!c^URR(a<[U4SjURR55(d Sn[U5egg)zEnsure the provided data dictionary contains all required label fields. Args: data (dict[str, Any]): The data dictionary to validate. Raises: ValueError: If any label field specified in params is missing from the data. c3,># UH oT;v M g7fr%r6).0ir3s r 7KeypointsProcessor.ensure_data_valid..s/\C[aT C[szaYour 'label_fields' are not valid - them must have same names as params in 'keypoint_params' dictN)rTr-all ValueError)r+r3msgs ` r ensure_data_valid$KeypointsProcessor.ensure_data_validsD ;; # #C/\4;;C[C[/\,\,\uCS/ !-] #r!cBU [XURRS9$)aFilter keypoints based on visibility within given shape. Args: data (np.ndarray): Keypoints in [x, y, z, angle, scale] format shape (ShapeType): Shape to check against as {'height': height, 'width': width, 'depth': depth} Returns: np.ndarray: Filtered keypoints )r()rrTr(r+r3shapes r filterKeypointsProcessor.filters dkk>Z>Z[[r!c[X5 g)zCheck if keypoints are valid within the given shape. Args: data (np.ndarray): Keypoints to validate. shape (ShapeType): Shape to check against. N)rrgs r checkKeypointsProcessor.checks  $r!cUR(dU$URn[UURUURUR S9$)aConvert keypoints from internal Albumentations format to the specified format. Args: data (np.ndarray): Keypoints in Albumentations format. shape (ShapeType): Shape information for validation. Returns: np.ndarray: Converted keypoints in the target format. check_validityr))sizerTrr,r(r)r+r3rhrTs r convert_from_albumentations.KeypointsProcessor.convert_from_albumentationssEyyK4  MM !22#44   r!cUR(dU$URn[UURUURUR S9$)a Convert keypoints from the specified format to internal Albumentations format. Args: data (np.ndarray): Keypoints in source format. shape (ShapeType): Shape information for validation. Returns: np.ndarray: Converted keypoints in Albumentations format. ro)rqrTrr,r(r)rrs r convert_to_albumentations,KeypointsProcessor.convert_to_albumentationssEyyK2  MM !22#44   r!r6r%)rTr rUzdict[str, str] | NonerG)r3rFrENone)r3 np.ndarrayrhr rEry)r3ryrhr rErx)rHrIrJrKrLr'propertyrYrdrirlrsrvrNrOrPs@r r r s 55 "\\\  \$%      6       r!r c ^^ ^^USUSp2SU;nTSS2S4TSS2S4pe[R"US:XS:-5Sn[R"US:Xb:-5Sn/n [[U5[U5-5HMn X;aU R STU SUS XZ35 X;dM.U R S TU SUS Xj35 MO U(a`TR SS :aMTSS2S 4mUSm[R"TS:TT:-5Sn U R UUU4S jU 55 TR SS :aXTSS2S 4m [R"T S:T S [R-:-5Sn U R U U4SjU 55 U (a[SRU 55eg)uCheck if keypoint coordinates are within valid ranges for the given shape. This function validates that: 1. All x-coordinates are within [0, width) 2. All y-coordinates are within [0, height) 3. If depth is provided in shape, z-coordinates are within [0, depth) 4. Angles are within the range [0, 2π) Args: keypoints (np.ndarray): Array of keypoints with shape (N, 5+), where N is the number of keypoints. - First 2 columns are always x, y - Column 3 (if present) is z - Column 4 (if present) is angle - Column 5+ (if present) are additional attributes shape (ShapeType): The shape of the image/volume: - For 2D: {'height': int, 'width': int} - For 3D: {'height': int, 'width': int, 'depth': int} Raises: ValueError: If any keypoint coordinate is outside the valid range, or if angles are invalid. The error message will detail which keypoints are invalid and why. Note: - The function assumes that keypoint coordinates are in absolute pixel values, not normalized - Angles are in radians - Z-coordinates are only checked if 'depth' is present in shape heightwidthdepthNrrzExpected x for keypoint  to be in range [0, ), got zExpected y for keypoint rc3F># UHnSTUSTSTU3v M g7f)zExpected z for keypoint rrNr6)r]idxr~rXzs r r_"check_keypoints..;s9 mvfi&y~&66J5'QXYZ[^Y_X` amvs!c3@># UHnSTUSTU3v M g7f)zExpected angle for keypoint u to be in range [0, 2π), got Nr6)r]rrrXs r r_rCs3 %+9S>*::XY_`cYdXe f%s ) rwheresortedsetappendrhextendmathrrbjoin)rXrhr|r} has_depthxy invalid_x invalid_yerror_messagesr invalid_zinvalid_anglesrr~rs` @@@r rrs:(OU7^E5 I QT?IadOq!a%AJ/03I!a%AK01!4INc)ns9~56    ! !*9S>*::NugU\]^]c\de     ! !*9S>*::NvhV]^_^d]ef  7Y__Q'!+ adOgHHa!eU 34Q7  mv  qA1a46A:&AK2G"HI!L %  >233r!cU(dU$UR(dU$USUSURSS5pTnUSS2S4USS2S4USS2S4pnUS:Xd:-US:-Xs:-n Ub XS:X:--n X $)aFilter keypoints to remove those outside the boundaries. Args: keypoints (np.ndarray): A numpy array of shape (N, 5+) where N is the number of keypoints. Each row represents a keypoint (x, y, z, angle, scale, ...). shape (ShapeType): Shape to check against as {'height': height, 'width': width, 'depth': depth}. remove_invisible (bool): If True, remove keypoints outside the boundaries. Returns: np.ndarray: Filtered keypoints. r|r}r~Nrrr)rqget) rXrhr(r|r}r~rrrvisibles r rrLs"  >> ?E'NEIIgtG Fqy))  r!c|U[;a[SUS[35e/SQ/SQ/SQ/SQ/SQ/SQ/S QS .nXQn[R"URS [ 4[R S 9n[U5HupU cM US S 2U 4US S 2U4'M U(a*USb$[R"US S 2S45US S 2S4'[US S 2S45US S 2S4'URS[U5:a)[R"XpS S 2[U5S 2445nU(a [Xr5 U$)u;Convert keypoints from various formats to the Albumentations format. This function takes keypoints in different formats and converts them to the standard Albumentations format: [x, y, z, angle, scale]. For 2D formats, z is set to 0. For formats without angle or scale, these values are set to 0. Args: keypoints (np.ndarray): Array of keypoints with shape (N, 2+), where N is the number of keypoints. The number of columns depends on the source_format. source_format (Literal["xy", "yx", "xya", "xys", "xyas", "xysa", "xyz"]): The format of the input keypoints. - "xy": [x, y] - "yx": [y, x] - "xya": [x, y, angle] - "xys": [x, y, scale] - "xyas": [x, y, angle, scale] - "xysa": [x, y, scale, angle] - "xyz": [x, y, z] shape (ShapeType): The shape of the image {'height': height, 'width': width, 'depth': depth}. check_validity (bool, optional): If True, check if the converted keypoints are within the image boundaries. Defaults to False. angle_in_degrees (bool, optional): If True, convert input angles from degrees to radians. Defaults to True. Returns: np.ndarray: Array of keypoints in Albumentations format [x, y, z, angle, scale] with shape (N, 5+). Any additional columns from the input keypoints are preserved and appended after the first 5 columns. Raises: ValueError: If the source_format is not one of the supported formats. Note: - For 2D formats (xy, yx, xya, xys, xyas, xysa), z coordinate is set to 0 - Angles are converted to the range [0, 2π) radians - If the input keypoints have additional columns beyond what's specified in the source_format, these columns are preserved in the output zUnknown source_format . Supported formats are: )rrNNN)rrNNN)rrNrN)rrNNr)rrNrr)rrNrr)rrrNNrrrrrrrr)dtypeNrr) keypoint_formatsrbrzerosrhrfloat32 enumerateradiansrlen column_stackr) rX source_formatrhrpr)format_to_indicesindicesprocessed_keypointsr^rs r rrpsEZ,,1-@YZjYklmm'&$$""$6!2 @G((IOOA$68_#`hjhrhrsG$ ?(1!S&(9 1 %%GAJ2$&JJ/B1a4/H$IAqD! 23Fq!t3L M1qC .. oo/BaQTUbQcQeNeDf.gh+3 r!c U[;a[SUS[35eUSS2S4USS2S4USS2S4USS2S4USS2S44upVpxn [U5nU(a#[[R "XVXxU 45U5 U(a[R "U5nXV/Xe/XVU/XVU /XVX/XVX/XVU/S .n [R "X5n URS[:a$[R "XSS2[S2445$U $) uConvert keypoints from Albumentations format to various other formats. This function takes keypoints in the standard Albumentations format [x, y, z, angle, scale] and converts them to the specified target format. Args: keypoints (np.ndarray): Array of keypoints in Albumentations format with shape (N, 5+), where N is the number of keypoints. Each row represents a keypoint [x, y, z, angle, scale, ...]. target_format (Literal["xy", "yx", "xya", "xys", "xyas", "xysa", "xyz"]): The desired output format. - "xy": [x, y] - "yx": [y, x] - "xya": [x, y, angle] - "xys": [x, y, scale] - "xyas": [x, y, angle, scale] - "xysa": [x, y, scale, angle] - "xyz": [x, y, z] shape (ShapeType): The shape of the image {'height': height, 'width': width, 'depth': depth}. check_validity (bool, optional): If True, check if the keypoints are within the image boundaries. Defaults to False. angle_in_degrees (bool, optional): If True, convert output angles to degrees. If False, angles remain in radians. Defaults to True. Returns: np.ndarray: Array of keypoints in the specified target format with shape (N, 2+). Any additional columns from the input keypoints beyond the first 5 are preserved and appended after the converted columns. Raises: ValueError: If the target_format is not one of the supported formats. Note: - Input angles are assumed to be in the range [0, 2π) radians - If the input keypoints have additional columns beyond the first 5, these columns are preserved in the output zUnknown target_format rNrrrrr) rrbrrrrdegreesrhr) rX target_formatrhrpr)rrranglescaleformat_to_columnsresults r rrs/Z,,1-@YZjYklmm%adOYq!t_i1oyYZ\]Y]`ijkmnjn`ooA!E u %Eq(?@%H 5!ffe}e}u$u$ay__.= >FqCC!5\5]2](^_`` Mr!)rryrEry)rXryrhr rErx)rXryrhr r(rDrEry)FT) rXryr8Literal['xy', 'yx', 'xya', 'xys', 'xyas', 'xysa', 'xyz']rhr rprDr)rDrEry) rXryrrrhr rprDr)rDrEry)rL __future__rrcollections.abcrtypingrrnumpyr$albumentations.core.type_definitionsrutilsr r r __all__rrr r rrrrr6r!r rs!# $X33 E % T VT nw w tE4P!! !! !P!! MMKM M M  M  Mh!! IIKI I I  I  Ir!