v1.0.8 - Change RGB Conversion's kernel to Spline36

rlaphoenix · Dec 19, 2019 · 4431c81 · 4431c81
1 parent e8bdbbc
commit 4431c81
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="vsgan",
-    version="1.0.7",
+    version="1.0.8",
     author="PRAGMA",
     author_email="[email protected]",
     description="VapourSynth GAN Implementation using RRDBNet, based on ESRGAN's implementation",

diff --git a/vsgan/__init__.py b/vsgan/__init__.py
@@ -40,10 +40,8 @@ def load_model(self, model, scale):
         self.rrdb_net_model = self.rrdb_net_model.to(self.torch_device)
 
     def run(self, clip, chunk=False):
-        # remember the clip's original format
-        original_format = clip.format
         # convert clip to RGB24 as it cannot read any other color space
-        buffer = mvsfunc.ToRGB(clip, depth=8)  # expecting RGB24 8bit
+        buffer = mvsfunc.ToRGB(clip, depth=8, kernel="spline36")  # expecting RGB24 (RGB 8bpp)
         # send the clip array to execute()
         results = []
         for c in self.chunk_clip(buffer) if chunk else [buffer]: