Fix possible bug in MotionBlur (#2168)

* fix near empty kernel when direction is -1 * update default value of direction_range
albumentations-team · Nov 29, 2024 · eaa1b5f · eaa1b5f
1 parent 96c10b5
commit eaa1b5f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/albumentations/augmentations/blur/functional.py b/albumentations/augmentations/blur/functional.py
@@ -196,7 +196,9 @@ def create_motion_kernel(
 
     # Apply direction bias
     if direction != 0:
-        t = t * (1 + direction)
+        t = t * (1 + abs(direction))
+        if direction < 0:
+            t = t * -1
 
     # Generate line coordinates
     x = center + dx * t

diff --git a/albumentations/augmentations/blur/transforms.py b/albumentations/augmentations/blur/transforms.py
@@ -225,7 +225,7 @@ def __init__(
         blur_limit: ScaleIntType = 7,
         allow_shifted: bool = True,
         angle_range: tuple[float, float] = (0, 360),
-        direction_range: tuple[float, float] = (-1.0, 1.0),
+        direction_range: tuple[float, float] = (-0.5, 0.5),
         always_apply: bool | None = None,
         p: float = 0.5,
     ):