| from PIL import ImageEnhance | |
| from .base import VideoProcessor | |
| class ContrastEditor(VideoProcessor): | |
| def __init__(self, rate=1.5): | |
| self.rate = rate | |
| 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 | |
| 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 | |