Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-02-27 02:34:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed waifu2x-caffe upscale by resolution errors and optimized upscaler code

Browse Source
This commit is contained in:
K4YT3X
2020-09-12 19:34:48 -04:00
parent 6dbdf93ca4
commit a82fcc778e
2 changed files with 55 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/wrappers/waifu2x_caffe.py
View File

@@ -4,7 +4,7 @@
Name: Waifu2x Caffe Driver
Author: K4YT3X
Date Created: Feb 24, 2018
Last Modified: September 9, 2020
Last Modified: September 12, 2020
Description: This class is a high-level wrapper
for waifu2x-caffe.
@@ -69,7 +69,14 @@ class WrapperMain:
# it will up updated later
self.driver_settings['output_depth'] = 12
def set_scale_resolution(self, width: int, height: int):
self.driver_settings['scale_width'] = width
self.driver_settings['scale_height'] = height
self.driver_settings['scale_ratio'] = None
def set_scale_ratio(self, scale_ratio: float):
self.driver_settings['scale_width'] = None
self.driver_settings['scale_height'] = None
self.driver_settings['scale_ratio'] = scale_ratio
def upscale(self, input_directory, output_directory):