OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2025-03-01 20:12:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added model arg

Browse Source
This commit is contained in:
K4YT3X 2018-04-20 13:00:57 -04:00
parent 254c761546
commit 56f654632e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
waifu2x.py
View File

@ -30,7 +30,7 @@ class WAIFU2X:
self.waifu2x_path = waifu2x_path
self.method = method
def upscale(self, folderin, folderout, width, height):
def upscale(self, folderin, folderout, width, height, model_type):
"""This is the core function for WAIFU2X class
[description]
@ -41,5 +41,5 @@ class WAIFU2X:
width {int} -- output video width
height {int} -- output video height
"""
os.system("{} -p {} -I png -i {} -e png -o {} -w {} -h {} -n 3 -m noise_scale -y photo".format(
self.waifu2x_path, self.method, folderin, folderout, width, height))
os.system("{} -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, model_type))