OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-28 06:59:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

reverting code back to batch waifu2x upscaling according to @VanTanev 's issue

Browse Source
This commit is contained in:
K4YT3X 2018-09-29 11:02:36 -04:00
parent 456b6c7e74
commit 4eb2c8e63e
2 changed files with 13 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
video2x.py
View File

@ -152,15 +152,8 @@ def video2x():
# Upscale images one by one using waifu2x
avalon.info('Starting to upscale extracted images')
for (dirpath, dirnames, filenames) in os.walk(FRAMES):
file_list = tqdm(filenames, ascii=True)
for file in file_list:
if file[-4:].lower() == '.png':
image_path = '{}\\{}'.format(dirpath, file)
file_list.set_description('Upscaling: {}'.format(file))
# avalon.dbgInfo('Upscaling: {}'.format(image_path))
w2.upscale(image_path, UPSCALED, args.width, args.height)
avalon.info('Extraction complete')
w2.upscale(FRAMES, UPSCALED, args.width, args.height)
avalon.info('Conversion complete')
# Frames to Video
avalon.info('Converting extracted frames into video')

27
waifu2x.py
View File

@ -28,20 +28,15 @@ class WAIFU2X:
self.method = method
self.model_type = model_type
def upscale(self, file, upscaled, width, height):
"""This is the core function for WAIFU2X class
def upscale(self, folderin, folderout, width, height):
"""This is the core function for WAIFU2X class
[description]
Arguments:
file {string} -- input image
upscaled {string} -- output folder path
width {int} -- output video width
height {int} -- output video height
model_type {string} -- model to use for upscaling
"""
file_id = file.split('extracted_')[-1].split('.png')[0]
output_file = '{}\\{}{}{}'.format(upscaled, 'extracted_', file_id, '.png')
execute = "{} -p {} -I png -i {} -e png -o {} -w {} -h {} -n 3 -m noise_scale -y {}".format(
self.waifu2x_path, self.method, file, output_file, width, height, self.model_type)
subprocess.call(execute, stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)
Arguments:
folderin {string} -- source folder path
folderout {string} -- output folder path
width {int} -- output video width
height {int} -- output video height
"""
execute = "{} -p {} -I png -i {} -e png -o {} -w {} -h {} -n 3 -m noise_scale -y {}".format(
self.waifu2x_path, self.method, folderin, folderout, width, height, self.model_type)
subprocess.call(execute)