diff --git a/src/video2x_gui.py b/src/video2x_gui.py index d97fa7c..b998070 100755 --- a/src/video2x_gui.py +++ b/src/video2x_gui.py @@ -45,10 +45,12 @@ Contact: [k4yt3x@k4yt3x.com](mailto:k4yt3x@k4yt3x.com)""" AVAILABLE_DRIVERS = { "Waifu2X Caffe": "waifu2x_caffe", "Waifu2X Converter CPP": "waifu2x_converter_cpp", - "Waifu2X NCNN Vulkan": "waifu2x_ncnn_vulkan", - "SRMD NCNN Vulkan": "srmd_ncnn_vulkan", - "RealSR NCNN Vulkan": "realsr_ncnn_vulkan", + "Waifu2X ncnn Vulkan": "waifu2x_ncnn_vulkan", + "SRMD ncnn Vulkan": "srmd_ncnn_vulkan", + "RealSR ncnn Vulkan": "realsr_ncnn_vulkan", "Anime4KCPP": "anime4kcpp", + "DAIN ncnn Vulkan": "dain_ncnn_vulkan", + "RIFE ncnn Vulkan": "rife_ncnn_vulkan", } # get current working directory before it is changed by drivers diff --git a/src/video2x_gui.ui b/src/video2x_gui.ui index 1828586..ce67e09 100644 --- a/src/video2x_gui.ui +++ b/src/video2x_gui.ui @@ -378,7 +378,7 @@ - Waifu2X NCNN Vulkan + Waifu2X ncnn Vulkan @@ -393,12 +393,12 @@ - SRMD NCNN Vulkan + SRMD ncnn Vulkan - RealSR NCNN Vulkan + RealSR ncnn Vulkan @@ -406,6 +406,21 @@ Anime4KCPP + + + DAIN ncnn Vulkan + + + + + CAIN ncnn Vulkan + + + + + RIFE ncnn Vulkan + + diff --git a/src/wrappers/realsr_ncnn_vulkan.py b/src/wrappers/realsr_ncnn_vulkan.py index 6d0e174..d75b9be 100755 --- a/src/wrappers/realsr_ncnn_vulkan.py +++ b/src/wrappers/realsr_ncnn_vulkan.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- """ -Name: RealSR NCNN Vulkan Driver +Name: RealSR ncnn Vulkan Driver Creator: K4YT3X Date Created: May 26, 2020 Last Modified: September 21, 2020 @@ -24,7 +24,7 @@ from avalon_framework import Avalon class WrapperMain: - """This class communicates with RealSR NCNN Vulkan engine + """This class communicates with RealSR ncnn Vulkan engine An object will be created for this class, containing information about the binary address and the processing method. When being called @@ -66,7 +66,7 @@ class WrapperMain: self.driver_settings["s"] = int(scale_ratio) def upscale(self, input_directory, output_directory): - """This is the core function for RealSR NCNN Vulkan class + """This is the core function for RealSR ncnn Vulkan class Arguments: input_directory {string} -- source directory path diff --git a/src/wrappers/srmd_ncnn_vulkan.py b/src/wrappers/srmd_ncnn_vulkan.py index 5d4c152..db6dfc2 100755 --- a/src/wrappers/srmd_ncnn_vulkan.py +++ b/src/wrappers/srmd_ncnn_vulkan.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- """ -Name: SRMD NCNN Vulkan Driver +Name: SRMD ncnn Vulkan Driver Creator: K4YT3X Date Created: April 26, 2020 Last Modified: September 21, 2020 @@ -24,7 +24,7 @@ from avalon_framework import Avalon class WrapperMain: - """This class communicates with SRMD NCNN Vulkan engine + """This class communicates with SRMD ncnn Vulkan engine An object will be created for this class, containing information about the binary address and the processing method. When being called diff --git a/src/wrappers/waifu2x_ncnn_vulkan.py b/src/wrappers/waifu2x_ncnn_vulkan.py index 9ad9834..bf087e1 100755 --- a/src/wrappers/waifu2x_ncnn_vulkan.py +++ b/src/wrappers/waifu2x_ncnn_vulkan.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- """ -Name: Waifu2x NCNN Vulkan Driver +Name: Waifu2x ncnn Vulkan Driver Creator: SAT3LL Date Created: June 26, 2019 Last Modified: May 11, 2020