Add random horizontal flip

README.md

@@ -256,6 +256,7 @@ Spatial-level transforms will simultaneously change both an input image as well
 | [RandomCrop](https://explore.albumentations.ai/transform/RandomCrop)                             | ✓     | ✓    | ✓      | ✓         |
 | [RandomCropFromBorders](https://explore.albumentations.ai/transform/RandomCropFromBorders)       | ✓     | ✓    | ✓      | ✓         |
 | [RandomGridShuffle](https://explore.albumentations.ai/transform/RandomGridShuffle)               | ✓     | ✓    | ✓      | ✓         |
+| [RandomHorizontalFlip](https://explore.albumentations.ai/transform/RandomHorizontalFlip)         | ✓     | ✓    | ✓      | ✓         |
 | [RandomResizedCrop](https://explore.albumentations.ai/transform/RandomResizedCrop)               | ✓     | ✓    | ✓      | ✓         |
 | [RandomRotate90](https://explore.albumentations.ai/transform/RandomRotate90)                     | ✓     | ✓    | ✓      | ✓         |
 | [RandomScale](https://explore.albumentations.ai/transform/RandomScale)                           | ✓     | ✓    | ✓      | ✓         |

albumentations/augmentations/tk/transform.py

@@ -1,13 +1,17 @@
+from __future__ import annotations
+
 from typing import Annotated
 from warnings import warn
 
 from pydantic import AfterValidator
 
+from albumentations.augmentations.geometric.transforms import HorizontalFlip
 from albumentations.augmentations.transforms import ImageCompression
 from albumentations.core.pydantic import check_0plus, nondecreasing
 from albumentations.core.transforms_interface import BaseTransformInitSchema
+from albumentations.core.types import Targets
 
-__all__ = ["RandomJPEG"]
+__all__ = ["RandomJPEG", "RandomHorizontalFlip"]
 
 
 class RandomJPEG(ImageCompression):
@@ -69,3 +73,56 @@ def __init__(
 
     def get_transform_init_args_names(self) -> tuple[str, ...]:
         return ("jpeg_quality",)
+
+
+class RandomHorizontalFlip(HorizontalFlip):
+    """Horizontally flip the input randomly with a given probability.
+
+    This transform is an alias for HorizontalFlip, provided for compatibility with
+    torchvision and Kornia APIs. For new code, it is recommended to use
+    albumentations.HorizontalFlip directly.
+
+    Args:
+        p (float): probability of applying the transform. Default: 0.5.
+
+    Targets:
+        image, mask, bboxes, keypoints
+
+    Image types:
+        uint8, float32
+
+    Number of channels:
+        Any
+
+    Note:
+        This is a direct alias for albumentations.HorizontalFlip transform.
+        It is provided to make migration from torchvision and Kornia easier by
+        maintaining API compatibility.
+
+    Example:
+        >>> transform = A.RandomHorizontalFlip(p=0.5)
+        >>> # Consider using instead:
+        >>> transform = A.HorizontalFlip(p=0.5)
+
+    References:
+        - torchvision: https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.RandomHorizontalFlip
+        - Kornia: https://kornia.readthedocs.io/en/latest/augmentation.html#kornia.augmentation.RandomHorizontalFlip
+    """
+
+    _targets = (Targets.IMAGE, Targets.MASK, Targets.BBOXES, Targets.KEYPOINTS)
+
+    class InitSchema(BaseTransformInitSchema):
+        pass
+
+    def __init__(
+        self,
+        p: float = 0.5,
+        always_apply: bool | None = None,
+    ):
+        warn(
+            "RandomHorizontalFlip is an alias for HorizontalFlip transform. "
+            "Consider using HorizontalFlip directly from albumentations.HorizontalFlip. ",
+            UserWarning,
+            stacklevel=2,
+        )
+        super().__init__(p=p, always_apply=always_apply)

tests/aug_definitions.py

@@ -383,4 +383,5 @@
     [A.TimeMasking, {"time_mask_param": 10}],
     [A.FrequencyMasking, {"freq_mask_param": 10}],
     [A.RandomJPEG, {"jpeg_quality": (50, 50)}],
+    [A.RandomHorizontalFlip, {}],
 ]

tests/test_transforms.py

@@ -2056,6 +2056,7 @@ def test_mask_dropout_bboxes(remove_invisible, expected_keypoints):
             A.Perspective,
             A.RandomGridShuffle,
             A.TimeReverse,
+            A.RandomHorizontalFlip,
         },
     ),
 )