fix(api): reuse results of blur modes

ssube · Jan 15, 2023 · ef06b45 · ef06b45
1 parent 1c2c8b2
commit ef06b45
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/api/onnx_web/image.py b/api/onnx_web/image.py
@@ -54,8 +54,7 @@ def mask_filter_gaussian(mask_image: Image, dims: Tuple[int, int], origin: Tuple
     noise.paste(mask_image, origin)
 
     for i in range(rounds):
-        blur = noise.copy()
-        blur.filter(ImageFilter.GaussianBlur(5))
+        blur = noise.filter(ImageFilter.GaussianBlur(5))
         noise = ImageChops.screen(noise, blur)
 
     return noise
@@ -81,7 +80,7 @@ def noise_source_gaussian(source_image: Image, dims: Tuple[int, int], origin: Tu
     noise.paste(source_image, origin)
 
     for i in range(rounds):
-        noise.filter(ImageFilter.GaussianBlur(5))
+        noise = noise.filter(ImageFilter.GaussianBlur(5))
 
     return noise