hugging-apps/echo-memory

1from PIL import ImageEnhance2from .base import VideoProcessor3 4 5class ContrastEditor(VideoProcessor):6    def __init__(self, rate=1.5):7        self.rate = rate8 9    @staticmethod10    def from_model_manager(model_manager, **kwargs):11        return ContrastEditor(**kwargs)12    13    def __call__(self, rendered_frames, **kwargs):14        rendered_frames = [ImageEnhance.Contrast(i).enhance(self.rate) for i in rendered_frames]15        return rendered_frames16 17 18class SharpnessEditor(VideoProcessor):19    def __init__(self, rate=1.5):20        self.rate = rate21 22    @staticmethod23    def from_model_manager(model_manager, **kwargs):24        return SharpnessEditor(**kwargs)25    26    def __call__(self, rendered_frames, **kwargs):27        rendered_frames = [ImageEnhance.Sharpness(i).enhance(self.rate) for i in rendered_frames]28        return rendered_frames29