From 1433ce54b16e56b5d94d7352daf1579001199fb5 Mon Sep 17 00:00:00 2001 From: Fatemeh Amerehi <107806531+f-amerehi@users.noreply.github.com> Date: Wed, 16 Aug 2023 13:04:01 +0330 Subject: [PATCH] Fix: update random_rain.py (#2514) --- kornia/augmentation/_2d/intensity/random_rain.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/kornia/augmentation/_2d/intensity/random_rain.py b/kornia/augmentation/_2d/intensity/random_rain.py index bbe5115d68..ea61e3de70 100644 --- a/kornia/augmentation/_2d/intensity/random_rain.py +++ b/kornia/augmentation/_2d/intensity/random_rain.py @@ -63,7 +63,7 @@ def apply_transform( bool(torch.all(torch.abs(params['drop_width_factor']) <= image.shape[3])), "Width of drop should be less than image width.", ) - + modeified_img = image.clone() for i in range(image.shape[0]): number_of_drops: int = int(params['number_of_drops_factor'][i]) # We generate tensor with maximum number of drops, and then remove unnecessary drops. @@ -87,5 +87,5 @@ def apply_transform( y = torch.linspace(start=0, end=width_of_drop, steps=size_of_line, dtype=torch.long).to(image.device) # Draw lines for k in range(x.shape[0]): - image[i, :, coords[0] + x[k], coords[1] + y[k]] = 200 / 255 - return image + modeified_img[i, :, coords[0] + x[k], coords[1] + y[k]] = 200 / 255 + return modeified_img