^j,UdZddlmZddlmZddlmZehdZde d<ehdZ de d <dd Z d d  dd Z ddZ ddZGddZ ddZy)z6Shared helpers for dataset augmentation configuration.) annotations)Callable)Any>D4FlipHorizontalFlipzfrozenset[str]HFLIP_TRANSFORM_NAMES>OneOfSomeOf SequentialCONTAINER_TRANSFORM_NAMESc|d||y)zBEmit the standard warning for a disabled keypoint horizontal flip.zKeypoint pipeline: '%s' performs a horizontal flip but no keypoint flip pairs were configured. The transform has been disabled to prevent incorrect keypoint annotations. Remove '%s' from your augmentation config or provide keypoint_flip_pairs.N)aug_namewarns e/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/rfdetr/datasets/_aug_utils.py_warn_keypoint_hflip_disabledrs a  T)include_keypointsc|s|St|tr t||St|tr t ||S|S)aDrop horizontal-flip transforms from keypoint augmentation configs. The helper preserves the input config shape: dictionary configs return dictionaries, and ordered list configs return lists. Container transforms are filtered recursively so nested hflip entries cannot survive inside ``OneOf``, ``SomeOf``, or ``Sequential``. Args: config: Augmentation config accepted by RF-DETR augmentation builders. include_keypoints: Whether the config will be applied to keypoint data. warn: Warning sink, typically ``logger.warning``. Returns: A filtered augmentation config with keypoint-unsafe hflip entries removed. r) isinstancelist_filter_keypoint_hflip_entriesdict_filter_keypoint_hflip_dict)configrrs r#filter_keypoint_hflip_augmentationsrs?*  &$-f4@@&$*6== Mrc&g}|D]}t|trt|dk7r|j|3t t |j \}}t|||}t|trw|j||i|S)z/Filter ordered single-key augmentation entries.r) rrlenappendnextiteritems_filter_keypoint_hflip_item_DropAugmentation)entriesrfiltered_entriesentryrparamsfiltered_paramss rrr@s-/ =%&#e*/  # #E * U[[] 34&5hTR o'8 9 ? ;< = rci}|jD])\}}t|||}t|tr%|||<+|S)z+Filter a mapping-style augmentation config.r)r%r&rr')rrfiltered_configrr+r,s rrrPsO&(O"LLN4&5hTR o'8 9 $3! 4 rceZdZdZy)r'z;Sentinel used when an augmentation entry should be removed.N)__name__ __module__ __qualname____doc__rrrr'r'[sErr'cD|tvrt||tS|tvr|St |t rt ||}|r|StSt |trd|vr|S|d}t |t s|St ||}|s tSi|d|iS)zCFilter one augmentation entry, recursing into supported containers.r transforms)r rr'r rrrr)rr+rfiltered_transformsr5s rr&r&_s((%h5 ""00 &$rCs ="$)22R(S~S,56W,X>X#     B  FF99 9  9  9r