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

reverting code back to batch waifu2x upscaling according to @VanTanev 's issue

Browse Source
This commit is contained in:
K4YT3X
2018-09-29 11:02:36 -04:00
parent 456b6c7e74
commit 4eb2c8e63e
2 changed files with 13 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
waifu2x.py
View File

@@ -28,20 +28,15 @@ class WAIFU2X:
self.method = method
self.model_type = model_type
def upscale(self, file, upscaled, width, height):
"""This is the core function for WAIFU2X class
def upscale(self, folderin, folderout, width, height):
"""This is the core function for WAIFU2X class
[description]
Arguments:
file {string} -- input image
upscaled {string} -- output folder path
width {int} -- output video width
height {int} -- output video height
model_type {string} -- model to use for upscaling
"""
file_id = file.split('extracted_')[-1].split('.png')[0]
output_file = '{}\\{}{}{}'.format(upscaled, 'extracted_', file_id, '.png')
execute = "{} -p {} -I png -i {} -e png -o {} -w {} -h {} -n 3 -m noise_scale -y {}".format(
self.waifu2x_path, self.method, file, output_file, width, height, self.model_type)
subprocess.call(execute, stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)
Arguments:
folderin {string} -- source folder path
folderout {string} -- output folder path
width {int} -- output video width
height {int} -- output video height
"""
execute = "{} -p {} -I png -i {} -e png -o {} -w {} -h {} -n 3 -m noise_scale -y {}".format(
self.waifu2x_path, self.method, folderin, folderout, width, height, self.model_type)
subprocess.call(execute)