| from PIL import ImageEnhance |
| from .base import VideoProcessor |
|
|
|
|
| class ContrastEditor(VideoProcessor): |
| def __init__(self, rate=1.5): |
| self.rate = rate |
|
|
| @staticmethod |
| def from_model_manager(model_manager, **kwargs): |
| return ContrastEditor(**kwargs) |
| |
| def __call__(self, rendered_frames, **kwargs): |
| rendered_frames = [ImageEnhance.Contrast(i).enhance(self.rate) for i in rendered_frames] |
| return rendered_frames |
|
|
|
|
| class SharpnessEditor(VideoProcessor): |
| def __init__(self, rate=1.5): |
| self.rate = rate |
|
|
| @staticmethod |
| def from_model_manager(model_manager, **kwargs): |
| return SharpnessEditor(**kwargs) |
| |
| def __call__(self, rendered_frames, **kwargs): |
| rendered_frames = [ImageEnhance.Sharpness(i).enhance(self.rate) for i in rendered_frames] |
| return rendered_frames |
|
|
