OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-29 16:09:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

renamed SRMD and RealSR classes per Python conventions

Browse Source
This commit is contained in:
K4YT3X 2021-08-17 14:43:49 +00:00
parent 736771b1c8
commit ba336e9c75
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
video2x/upscaler.py
View File

@ -4,13 +4,13 @@
Name: Upscaler Name: Upscaler
Author: K4YT3X Author: K4YT3X
Date Created: May 27, 2021 Date Created: May 27, 2021
Last Modified: June 28, 2021 Last Modified: August 17, 2021
""" """
# local imports # local imports
from .wrappers.realsr_ncnn_vulkan_python.realsr_ncnn_vulkan import RealSR from realsr_ncnn_vulkan_python.realsr_ncnn_vulkan import Realsr
from .wrappers.srmd_ncnn_vulkan_python.srmd_ncnn_vulkan import SRMD from srmd_ncnn_vulkan_python.srmd_ncnn_vulkan import Srmd
from .wrappers.waifu2x_ncnn_vulkan_python.waifu2x_ncnn_vulkan import Waifu2x from waifu2x_ncnn_vulkan_python.waifu2x_ncnn_vulkan import Waifu2x
# built-in imports # built-in imports
import math import math
@ -33,7 +33,7 @@ DRIVER_FIXED_SCALING_RATIOS = {
"realsr": [4], "realsr": [4],
} }
DRIVER_CLASSES = {"waifu2x": Waifu2x, "srmd": SRMD, "realsr": RealSR} DRIVER_CLASSES = {"waifu2x": Waifu2x, "srmd": Srmd, "realsr": Realsr}
class Upscaler(multiprocessing.Process): class Upscaler(multiprocessing.Process):
@ -174,5 +174,5 @@ class Upscaler(multiprocessing.Process):
self.running = False self.running = False
return super().run() return super().run()
def _stop(self, signal_number, frame): def _stop(self, _signal_number, _frame):
self.running = False self.running = False