From 4431c81c728c8b61fef9ab5bcb456b5eb14d62a9 Mon Sep 17 00:00:00 2001 From: PRAGMA <17136956+imPRAGMA@users.noreply.github.com> Date: Thu, 19 Dec 2019 06:25:55 +0000 Subject: [PATCH] v1.0.8 - Change RGB Conversion's kernel to Spline36 --- setup.py | 2 +- vsgan/__init__.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 6aeaa7c..f231126 100644 --- 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="pragma.exe@gmail.com", description="VapourSynth GAN Implementation using RRDBNet, based on ESRGAN's implementation", diff --git a/vsgan/__init__.py b/vsgan/__init__.py index 20fb22d..4cd97e2 100644 --- 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]: