update smoothers

diffsynth/processors/PILEditor.py

@@ -0,0 +1,28 @@
+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