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

2.8.1 added automatic pixel format and color depth detection

Browse Source
This commit is contained in:
k4yt3x
2019-07-09 20:47:55 -04:00
parent 8b845e35b3
commit ca90c5be02
5 changed files with 62 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bin/waifu2x_caffe.py
View File

@@ -6,7 +6,7 @@
Name: Waifu2x Caffe Driver
Author: K4YT3X
Date Created: Feb 24, 2018
Last Modified: June 15, 2019
Last Modified: July 9, 2019
Description: This class is a high-level wrapper
for waifu2x-caffe.
@@ -25,10 +25,11 @@ class Waifu2xCaffe:
the upscale function.
"""
def __init__(self, waifu2x_settings, process, model_dir):
def __init__(self, waifu2x_settings, process, model_dir, bit_depth):
self.waifu2x_settings = waifu2x_settings
self.waifu2x_settings['process'] = process
self.waifu2x_settings['model_dir'] = model_dir
self.waifu2x_settings['output_depth'] = bit_depth
# arguments passed through command line overwrites config file values
self.process = process