generated from thinkode/modelRepository
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from dataclasses import dataclass
|
|
|
|
import numpy as np
|
|
from PIL import Image, ImageDraw, ImageFilter
|
|
|
|
from moviepy.Clip import Clip
|
|
from moviepy.Effect import Effect
|
|
|
|
|
|
@dataclass
|
|
class HeadBlur(Effect):
|
|
"""Returns a filter that will blur a moving part (a head ?) of the frames.
|
|
|
|
The position of the blur at time t is defined by (fx(t), fy(t)), the radius
|
|
of the blurring by ``radius`` and the intensity of the blurring by ``intensity``.
|
|
"""
|
|
|
|
fx: callable
|
|
fy: callable
|
|
radius: float
|
|
intensity: float = None
|
|
|
|
def apply(self, clip: Clip) -> Clip:
|
|
"""Apply the effect to the clip."""
|
|
if self.intensity is None:
|
|
self.intensity = int(2 * self.radius / 3)
|
|
|
|
def filter(gf, t):
|
|
im = gf(t).copy()
|
|
h, w, d = im.shape
|
|
x, y = int(self.fx(t)), int(self.fy(t))
|
|
x1, x2 = max(0, x - self.radius), min(x + self.radius, w)
|
|
y1, y2 = max(0, y - self.radius), min(y + self.radius, h)
|
|
|
|
image = Image.fromarray(im)
|
|
mask = Image.new("RGB", image.size)
|
|
draw = ImageDraw.Draw(mask)
|
|
draw.ellipse([x1, y1, x2, y2], fill=(255, 255, 255))
|
|
|
|
blurred = image.filter(ImageFilter.GaussianBlur(radius=self.intensity))
|
|
|
|
res = np.where(np.array(mask) > 0, np.array(blurred), np.array(image))
|
|
return res
|
|
|
|
return clip.transform(filter)
|