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]: