update waifu2x_ncnn_vulkan wrapper for directory as input

add workaround for the double file extension
This commit is contained in:
sat3ll 2019-07-12 23:12:08 +01:00
parent 9429b04528
commit 40a4d3a602
3 changed files with 39 additions and 28 deletions

View File

@ -150,6 +150,14 @@ class Ffmpeg:
# read FFmpeg input options
execute.extend(self._read_configuration(phase='frames_to_video', section='input_options'))
# WORKAROUND FOR WAIFU2X-NCNN-VULKAN
import re
import shutil
regex = re.compile(r'\.png\.png$')
for raw_frame in os.listdir(upscaled_frames):
shutil.move(os.path.join(upscaled_frames, raw_frame), os.path.join(upscaled_frames, regex.sub('.png', raw_frame)))
# END WORKAROUND
# append input frames path into command
execute.extend([
'-i',

View File

@ -47,7 +47,9 @@
"scale-ratio": null,
"tile-size": 400,
"model-path": null,
"gpu": 0
"gpu": 0,
"load-proc-save_threads": null,
"verbose": null
},
"ffmpeg": {
"ffmpeg_path": "C:\\Users\\K4YT3X\\AppData\\Local\\video2x\\ffmpeg-latest-win64-static\\bin",

View File

@ -67,10 +67,11 @@ class Waifu2xNcnnVulkan:
'scale-ratio': '-s',
'tile-size': '-t',
'model-path': '-m',
'gpu': '-g'
'gpu': '-g',
'load-proc-save_threads': '-j',
'verbose': '-v'
}
for raw_frame in os.listdir(input_directory):
execute = [self.waifu2x_settings['waifu2x_ncnn_vulkan_path']]
for key in self.waifu2x_settings.keys():
value = self.waifu2x_settings[key]
@ -78,10 +79,10 @@ class Waifu2xNcnnVulkan:
continue
elif key == 'input':
execute.append(waifu2x_ncnn_vulkan_opt_flag[key])
execute.append(os.path.join(input_directory, raw_frame))
execute.append(input_directory)
elif key == 'output':
execute.append(waifu2x_ncnn_vulkan_opt_flag[key])
execute.append(os.path.join(output_directory, raw_frame))
execute.append(output_directory)
elif key == 'scale-ratio':
execute.append(waifu2x_ncnn_vulkan_opt_flag[key])
# waifu2x_ncnn_vulkan does not accept an arbitrary scale ratio, max is 2