OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2025-03-03 13:02:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added function to absolutify paths set in config file for better exe support

Browse Source
This commit is contained in:
k4yt3x 2019-03-31 03:23:08 -04:00
parent 4973ae5b8d
commit 25156077f2
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
bin/video2x.py
View File

@ -45,6 +45,7 @@ import GPUtil
import json import json
import os import os
import psutil import psutil
import re
import shutil import shutil
import sys import sys
import tempfile import tempfile
@ -170,6 +171,40 @@ def read_config(config_file):
return config return config
def absolutify_paths(config):
""" Check to see if paths to binaries are absolute
This function checks if paths to binary files are absolute.
If not, then absolutify the path.
Arguments:
config {dict} -- configuration file dictionary
Returns:
dict -- configuration file dictionary
"""
current_folder = os.path.dirname(os.path.abspath(sys.argv[0]))
# check waifu2x-caffe path
if not re.match('^[a-z]:', config['waifu2x_caffe']['waifu2x_caffe_path'], re.IGNORECASE):
config['waifu2x_caffe']['waifu2x_caffe_path'] = '{}\\{}'.format(current_folder, config['waifu2x_caffe']['waifu2x_caffe_path'])
# check waifu2x-converter-cpp path
if not re.match('^[a-z]:', config['waifu2x_converter']['waifu2x_converter_path'], re.IGNORECASE):
config['waifu2x_converter']['waifu2x_converter_path'] = '{}\\{}'.format(current_folder, config['waifu2x_converter']['waifu2x_converter_path'])
# check ffmpeg path
if not re.match('^[a-z]:', config['ffmpeg']['ffmpeg_path'], re.IGNORECASE):
config['ffmpeg']['ffmpeg_path'] = '{}\\{}'.format(current_folder, config['ffmpeg']['ffmpeg_path'])
# check video2x cache path
if config['video2x']['video2x_cache_folder']:
if not re.match('^[a-z]:', config['video2x']['video2x_cache_folder'], re.IGNORECASE):
config['video2x']['video2x_cache_folder'] = '{}\\{}'.format(current_folder, config['video2x']['video2x_cache_folder'])
return config
# /////////////////// Execution /////////////////// # # /////////////////// Execution /////////////////// #
# this is not a library # this is not a library
@ -198,6 +233,7 @@ check_memory()
# read configurations from JSON # read configurations from JSON
config = read_config(args.config) config = read_config(args.config)
config = absolutify_paths(config)
# load waifu2x configuration # load waifu2x configuration
if args.driver == 'waifu2x_caffe': if args.driver == 'waifu2x_caffe':
@ -249,6 +285,7 @@ try:
# start timer # start timer
begin_time = time.time() begin_time = time.time()
# if input specified is a single file
if os.path.isfile(args.input): if os.path.isfile(args.input):
""" Upscale single video file """ """ Upscale single video file """
Avalon.info('Upscaling single video file: {}'.format(args.input)) Avalon.info('Upscaling single video file: {}'.format(args.input))
@ -268,6 +305,8 @@ try:
# run upscaler- # run upscaler-
upscaler.run() upscaler.run()
upscaler.cleanup() upscaler.cleanup()
# if input specified is a folder
elif os.path.isdir(args.input): elif os.path.isdir(args.input):
""" Upscale videos in a folder/directory """ """ Upscale videos in a folder/directory """
Avalon.info('Upscaling videos in folder/directory: {}'.format(args.input)) Avalon.info('Upscaling videos in folder/directory: {}'.format(args.input))