bugfix: motion blur works on arbitrarily small images

bethgelab · Oct 30, 2019 · 94aaac4 · 94aaac4
1 parent 4acffd8
commit 94aaac4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/imagecorruptions/corruptions.py b/imagecorruptions/corruptions.py
@@ -143,12 +143,13 @@ def _motion_blur(x, radius, sigma, angle):
     for i in range(width):
         dx = -math.ceil(((i*point[0]) / hypot) - 0.5)
         dy = -math.ceil(((i*point[1]) / hypot) - 0.5)
+        if (np.abs(dy) >= x.shape[0] or np.abs(dx) >= x.shape[1]):
+            # simulated motion exceeded image borders
+            break
         shifted = shift(x, dx, dy)
         blurred = blurred + kernel[i] * shifted
-
     return blurred
 
-
 # /////////////// End Corruption Helpers ///////////////