basic functions online

2026-02-21 14:24:53 +08:00 · 2018-02-24 16:13:27 -05:00
parent 3929259d36
commit 57733d602b
4 changed files with 81 additions and 19 deletions
--- a/waifu2x.py
+++ b/waifu2x.py
@@ -17,8 +17,8 @@ import os

 class WAIFU2X:

-    def __init__(self):
-        pass
+    def __init__(self, waifu2x_path):
+        self.waifu2x_path = waifu2x_path

-    def upscale(self, factor, folderin, folderout):
-        os.system("waifu2x-caffe-cui.exe -i {} -o {}".format(folderin, folderout))
+    def upscale(self, folderin, folderout, width, height):
+        os.system("{} -p cpu -I png -i {} -e png -o {} -w {} -h {}".format(self.waifu2x_path, folderin, folderout, width, height))