[Dy2St] RandomHorizontalFlip

python/paddle/tests/test_transforms_static.py

@@ -100,6 +100,16 @@ def set_trans_api(self):
         self.api = transforms.RandomVerticalFlip(prob=1)
 
 
+class TestRandomHorizontalFlip0(TestTransformUnitTestBase):
+    def set_trans_api(self):
+        self.api = transforms.RandomHorizontalFlip(0)
+
+
+class TestRandomHorizontalFlip1(TestTransformUnitTestBase):
+    def set_trans_api(self):
+        self.api = transforms.RandomHorizontalFlip(1)
+
+
 class TestRandomCrop_random(TestTransformUnitTestBase):
     def get_shape(self):
         return (3, 240, 240)

python/paddle/vision/transforms/transforms.py

@@ -611,10 +611,23 @@ def __init__(self, prob=0.5, keys=None):
         self.prob = prob
 
     def _apply_image(self, img):
+        if paddle.in_dynamic_mode():
+            return self._dynamic_apply_image(img)
+        else:
+            return self._static_apply_image(img)
+
+    def _dynamic_apply_image(self, img):
         if random.random() < self.prob:
             return F.hflip(img)
         return img
 
+    def _static_apply_image(self, img):
+        return paddle.static.nn.cond(
+            paddle.rand(shape=(1,)) < self.prob,
+            lambda: F.hflip(img),
+            lambda: img,
+        )
+
 
 class RandomVerticalFlip(BaseTransform):
     """Vertically flip the input data randomly with a given probability.