Cleanup

albumentations-team · Jan 29, 2025 · 7a6bfbc · 7a6bfbc
1 parent 21bebb0
commit 7a6bfbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/albumentations/augmentations/functional.py b/albumentations/augmentations/functional.py
@@ -2716,7 +2716,7 @@ def apply_linear_illumination(img: np.ndarray, intensity: float, angle: float) -
 
     # Add channel dimension if needed
     if img.ndim == NUM_MULTI_CHANNEL_DIMENSIONS:
-        gradient = np.repeat(gradient[..., None], img.shape[2], axis=2)
+        gradient = cv2.merge([gradient] * img.shape[2])
 
     return albucore.multiply_by_array(img, gradient)