Upload Blur.py

Blur.py

@@ -0,0 +1,51 @@
+import cv2
+from PIL import Image
+import numpy as np
+from rembg import remove
+
+def cv_to_pil(img):
+	return Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGRA2RGBA))
+
+
+def pil_to_cv(img):
+	return cv2.cvtColor(np.array(img), cv2.COLOR_RGBA2BGRA)
+
+
+def motion_blur(img, distance, amount):
+    # Convert to RGBA
+    img = img.convert('RGBA')
+    
+    # Convert pil to cv
+    cv_img = pil_to_cv(img)
+
+    # Generating the kernel
+    kernel_motion_blur = np.zeros((distance, distance))
+    kernel_motion_blur[int((distance-1)/2), :] = np.ones(distance)
+    kernel_motion_blur = kernel_motion_blur / distance
+
+    # Applying the kernel to the input image
+    output = cv2.filter2D(cv_img, -1, kernel_motion_blur)
+
+    # Convert cv to pil
+    blur_img = cv_to_pil(output).convert('RGBA')
+
+    # Blend the original image and the blur image
+    final_img = Image.blend(img, blur_img, amount)
+
+    return final_img
+
+
+def background_motion_blur(background, distance_blur, amount_blur, amount_subject):
+    # Remove background
+    subject = remove(background)
+
+    # Blur the background
+    background_blur = motion_blur(background, distance_blur, amount_blur)
+    
+    # Put the subject on top of the blur background
+    subject_on_blur_background = Image.alpha_composite(background_blur, subject)
+
+    # Blend the subject and the blur background
+    result = Image.blend(background_blur, subject_on_blur_background, amount_subject)
+
+    return result