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

added support for JPEG and other image format

Browse Source
This commit is contained in:
k4yt3x
2019-03-30 14:01:36 -04:00
parent 562aaee056
commit 78ae2e7ed2
3 changed files with 23 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bin/waifu2x_caffe.py
View File

@@ -4,7 +4,7 @@
Name: Waifu2x Caffe Driver
Author: K4YT3X
Date Created: Feb 24, 2018
Last Modified: March 24, 2019
Last Modified: March 30, 2019
Description: This class is a high-level wrapper
for waifu2x-caffe.
@@ -33,7 +33,7 @@ class Waifu2xCaffe:
self.model_dir = model_dir
self.print_lock = threading.Lock()
def upscale(self, input_folder, output_folder, scale_ratio, scale_width, scale_height, upscaler_exceptions):
def upscale(self, input_folder, output_folder, scale_ratio, scale_width, scale_height, image_format, upscaler_exceptions):
"""This is the core function for WAIFU2X class
Arguments:
@@ -54,6 +54,8 @@ class Waifu2xCaffe:
self.waifu2x_settings['scale_width'] = scale_width
self.waifu2x_settings['scale_height'] = scale_height
self.waifu2x_settings['output_extention'] = image_format
# print thread start message
self.print_lock.acquire()
Avalon.debug_info('[upscaler] Thread {} started'.format(threading.current_thread().name))