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