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

made custom intermediate frame format available for waifu2x/srmd/realsr

Browse Source
This commit is contained in:
K4YT3X
2020-09-21 15:14:00 -04:00
parent 7de9cc699a
commit a0522311e2
4 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/wrappers/realsr_ncnn_vulkan.py
View File

@@ -4,7 +4,7 @@
Name: RealSR NCNN Vulkan Driver
Creator: K4YT3X
Date Created: May 26, 2020
Last Modified: September 9, 2020
Last Modified: September 21, 2020
Description: This class is a high-level wrapper
for realsr_ncnn_vulkan.
@@ -49,11 +49,13 @@ class WrapperMain:
parser.add_argument('-g', type=int, help='gpu device to use')
parser.add_argument('-j', type=str, help='thread count for load/proc/save')
parser.add_argument('-x', action='store_true', help='enable tta mode')
parser.add_argument('-f', type=str, help=argparse.SUPPRESS) # help='output image format (jpg/png/webp, default=ext/png)')
return parser.parse_args(arguments)
def load_configurations(self, upscaler):
# self.driver_settings['s'] = int(upscaler.scale_ratio)
self.driver_settings['j'] = '{}:{}:{}'.format(upscaler.processes, upscaler.processes, upscaler.processes)
self.driver_settings['f'] = upscaler.extracted_frame_format.lower()
def set_scale_ratio(self, scale_ratio: int):
self.driver_settings['s'] = int(scale_ratio)

4
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: September 9, 2020
Last Modified: September 21, 2020
Description: This class is a high-level wrapper
for srmd_ncnn_vulkan.
@@ -50,11 +50,13 @@ class WrapperMain:
parser.add_argument('-g', type=int, help='gpu device to use')
parser.add_argument('-j', type=str, help='thread count for load/proc/save')
parser.add_argument('-x', action='store_true', help='enable tta mode')
parser.add_argument('-f', type=str, help=argparse.SUPPRESS) # help='output image format (jpg/png/webp, default=ext/png)')
return parser.parse_args(arguments)
def load_configurations(self, upscaler):
# self.driver_settings['s'] = int(upscaler.scale_ratio)
self.driver_settings['j'] = '{}:{}:{}'.format(upscaler.processes, upscaler.processes, upscaler.processes)
self.driver_settings['f'] = upscaler.extracted_frame_format.lower()
def set_scale_ratio(self, scale_ratio: int):
self.driver_settings['s'] = int(scale_ratio)

5
src/wrappers/waifu2x_ncnn_vulkan.py
View File

@@ -7,7 +7,7 @@ Date Created: June 26, 2019
Last Modified: May 11, 2020
Editor: K4YT3X
Last Modified: September 9, 2020
Last Modified: September 21, 2020
Description: This class is a high-level wrapper
for waifu2x_ncnn_vulkan.
@@ -53,12 +53,13 @@ class WrapperMain:
parser.add_argument('-g', type=int, help='gpu device to use')
parser.add_argument('-j', type=str, help='thread count for load/proc/save')
parser.add_argument('-x', action='store_true', help='enable tta mode')
parser.add_argument('-f', type=str, help='output image format (jpg/png/webp, default=ext/png)')
parser.add_argument('-f', type=str, help=argparse.SUPPRESS) # help='output image format (jpg/png/webp, default=ext/png)')
return parser.parse_args(arguments)
def load_configurations(self, upscaler):
# self.driver_settings['s'] = int(upscaler.scale_ratio)
self.driver_settings['j'] = '{}:{}:{}'.format(upscaler.processes, upscaler.processes, upscaler.processes)
self.driver_settings['f'] = upscaler.extracted_frame_format.lower()
def set_scale_ratio(self, scale_ratio: int):
self.driver_settings['s'] = int(scale_ratio)