mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-19 23:08:13 +00:00
29 lines
855 B
Python
29 lines
855 B
Python
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
|