Update rotate.py (#2330)

albumentations-team · Feb 4, 2025 · 752264b · 752264b
1 parent 2764cb4
commit 752264b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/albumentations/augmentations/geometric/rotate.py b/albumentations/augmentations/geometric/rotate.py
@@ -136,7 +136,7 @@ class Rotate(DualTransform):
             Default: cv2.INTER_LINEAR.
         border_mode (OpenCV flag): Flag that is used to specify the pixel extrapolation method. Should be one of:
             cv2.BORDER_CONSTANT, cv2.BORDER_REPLICATE, cv2.BORDER_REFLECT, cv2.BORDER_WRAP, cv2.BORDER_REFLECT_101.
-            Default: cv2.BORDER_REFLECT_101
+            Default: cv2.BORDER_CONSTANT
         fill (ColorType): Padding value if border_mode is cv2.BORDER_CONSTANT.
         fill_mask (ColorType): Padding value if border_mode is cv2.BORDER_CONSTANT applied for masks.
         rotate_method (str): Method to rotate bounding boxes. Should be 'largest_box' or 'ellipse'.