Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-05-13 04:47:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

change waifu2x-ncnn-vulkan to underscores

Browse Source 
agglomerate waifu2x choices to a list
This commit is contained in:
sat3ll
2019-06-26 00:50:29 +01:00
parent 4e3565b3ac
commit 0b99ac33a6
5 changed files with 33 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bin/upscaler.py
View File

@@ -150,7 +150,8 @@ class Upscaler:
self.upscaler_exceptions = []
# initialize waifu2x driver
if self.waifu2x_driver != 'waifu2x_caffe' and self.waifu2x_driver != 'waifu2x_converter' and self.waifu2x_driver != 'waifu2x-ncnn-vulkan':
drivers = ['waifu2x_caffe', 'waifu2x_converter', 'waifu2x_ncnn_vulkan']
if self.waifu2x_driver not in drivers:
raise Exception(f'Unrecognized waifu2x driver: {self.waifu2x_driver}')
# it's easier to do multi-threading with waifu2x_converter
@@ -247,7 +248,7 @@ class Upscaler:
if len(self.upscaler_exceptions) != 0:
raise(self.upscaler_exceptions[0])
elif self.waifu2x_driver == 'waifu2x-ncnn-vulkan':
elif self.waifu2x_driver == 'waifu2x_ncnn_vulkan':
# create a container for all upscaler threads
upscaler_threads = []