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

yielding multithreading control to srmd/waifu2x-ncnn-vulkan, fixing FFmpeg framerate detection

Browse Source
This commit is contained in:
k4yt3x
2020-05-09 04:54:28 -04:00
parent e82a26d44f
commit bcb54b6d2c
3 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/wrappers/srmd_ncnn_vulkan.py
View File

@@ -4,7 +4,7 @@
Name: SRMD NCNN Vulkan Driver
Creator: K4YT3X
Date Created: April 26, 2020
Last Modified: May 7, 2020
Last Modified: May 9, 2020
Description: This class is a high-level wrapper
for srmd_ncnn_vulkan.
@@ -53,7 +53,7 @@ class WrapperMain:
parser.add_argument('-x', action='store_true', help='enable tta mode')
return parser.parse_args(arguments)
def upscale(self, input_directory, output_directory, scale_ratio):
def upscale(self, input_directory, output_directory, scale_ratio, threads):
"""This is the core function for SRMD ncnn Vulkan class
Arguments:
@@ -66,6 +66,7 @@ class WrapperMain:
self.driver_settings['i'] = input_directory
self.driver_settings['o'] = output_directory
self.driver_settings['s'] = scale_ratio
self.driver_settings['j'] = '{}:{}:{}'.format(threads, threads, threads)
# by default, srmd-ncnn-vulkan will look for the models under the current working directory
# change the working directory to its containing folder if model directory not specified

5
src/wrappers/waifu2x_ncnn_vulkan.py
View File

@@ -4,7 +4,7 @@
Name: Waifu2x NCNN Vulkan Driver
Creator: SAT3LL
Date Created: June 26, 2019
Last Modified: May 7, 2020
Last Modified: May 9, 2020
Editor: K4YT3X
Last Modified: February 22, 2020
@@ -56,7 +56,7 @@ class WrapperMain:
parser.add_argument('-x', action='store_true', help='enable tta mode')
return parser.parse_args(arguments)
def upscale(self, input_directory, output_directory, scale_ratio):
def upscale(self, input_directory, output_directory, scale_ratio, threads):
"""This is the core function for WAIFU2X class
Arguments:
@@ -69,6 +69,7 @@ class WrapperMain:
self.driver_settings['i'] = input_directory
self.driver_settings['o'] = output_directory
self.driver_settings['s'] = int(scale_ratio)
self.driver_settings['j'] = '{}:{}:{}'.format(threads, threads, threads)
# by default, waifu2x-ncnn-vulkan will look for the models under the current working directory
# change the working directory to its containing folder if model directory not specified