OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-29 16:09:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

unified comment styles, removed redundant information

Browse Source
This commit is contained in:
k4yt3x 2019-03-19 13:08:49 -04:00
parent 254a158e7f
commit 72d64a469b
1 changed files with 12 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
bin/waifu2x_caffe.py Executable file → Normal file
View File

@ -4,10 +4,10 @@
Name: Waifu2x Caffe Driver Name: Waifu2x Caffe Driver
Author: K4YT3X Author: K4YT3X
Date Created: Feb 24, 2018 Date Created: Feb 24, 2018
Last Modified: March 9, 2019 Last Modified: March 19, 2019
Description: This class controls waifu2x Description: This class is a high-level wrapper
engine for waifu2x-caffe.
""" """
from avalon_framework import Avalon from avalon_framework import Avalon
import subprocess import subprocess
@ -53,28 +53,23 @@ class Waifu2xCaffe:
self.waifu2x_settings['scale_width'] = scale_width self.waifu2x_settings['scale_width'] = scale_width
self.waifu2x_settings['scale_height'] = scale_height self.waifu2x_settings['scale_height'] = scale_height
# Print thread start message # print thread start message
self.print_lock.acquire() self.print_lock.acquire()
Avalon.debug_info('[upscaler] Thread {} started'.format(threading.current_thread().name)) Avalon.debug_info('[upscaler] Thread {} started'.format(threading.current_thread().name))
self.print_lock.release() self.print_lock.release()
""" # list to be executed
# Create string for execution
execute = '\"{}\" -p {} -I png -i \"{}\" -e png -o {} -w {} -h {} -n 3 -m noise_scale -y {}'.format(
self.waifu2x_path, self.process, input_folder, output_folder, width, height, self.model_dir)
"""
execute = [] execute = []
for key in self.waifu2x_settings.keys(): for key in self.waifu2x_settings.keys():
value = self.waifu2x_settings[key] value = self.waifu2x_settings[key]
# The key doesn't need to be passed in this case # the key doesn't need to be passed in this case
if key == 'waifu2x_caffe_path': if key == 'waifu2x_caffe_path':
execute.append(str(value)) execute.append(str(value))
# Null or None means that leave this option out (keep default) # null or None means that leave this option out (keep default)
elif value is None or value is False: elif value is None or value is False:
continue continue
@ -86,9 +81,12 @@ class Waifu2xCaffe:
execute.append(str(value)) execute.append(str(value))
Avalon.debug_info('Executing: {}'.format(execute)) Avalon.debug_info('Executing: {}'.format(execute))
subprocess.run(execute, check=True) completed_command = subprocess.run(execute, check=True)
# Print thread exiting message # print thread exiting message
self.print_lock.acquire() self.print_lock.acquire()
Avalon.debug_info('[upscaler] Thread {} exiting'.format(threading.current_thread().name)) Avalon.debug_info('[upscaler] Thread {} exiting'.format(threading.current_thread().name))
self.print_lock.release() self.print_lock.release()
# return command execution return code
return completed_command.returncode