OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2025-01-04 04:39:10 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

enhanced logo-printing

Browse Source
This commit is contained in:
k4yt3x 2019-08-25 23:03:18 -04:00
parent a685e91b14
commit 6521e071ce
1 changed files with 7 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
bin/video2x.py
View File

@ -13,7 +13,7 @@ __ __ _ _ ___ __ __
Name: Video2X Controller Name: Video2X Controller
Author: K4YT3X Author: K4YT3X
Date Created: Feb 24, 2018 Date Created: Feb 24, 2018
Last Modified: August 17, 2019 Last Modified: August 21, 2019
Dev: BrianPetkovsek Dev: BrianPetkovsek
Dev: SAT3LL Dev: SAT3LL
@ -49,9 +49,8 @@ from upscaler import Upscaler
# built-in imports # built-in imports
import argparse import argparse
import glob import contextlib
import json import json
import os
import pathlib import pathlib
import re import re
import shutil import shutil
@ -128,9 +127,8 @@ def process_arguments():
def print_logo(): def print_logo():
"""print video2x logo""" """print video2x logo"""
print(LOGO) print(LOGO)
print('\n Video2X Video Enlarger') print(f'\n{"Video2X Video Enlarger".rjust(40, " ")}')
spaces = ((44 - len(f'Version {VERSION}')) // 2) * ' ' print(f'\n{Avalon.FM.BD}{f"Version {VERSION}".rjust(36, " ")}{Avalon.FM.RST}\n')
print(f'{Avalon.FM.BD}\n{spaces} Version {VERSION}\n{Avalon.FM.RST}')
def check_memory(): def check_memory():
@ -153,12 +151,10 @@ def check_memory():
Avalon.warning('Nvidia-smi not available, skipping available memory check') Avalon.warning('Nvidia-smi not available, skipping available memory check')
Avalon.warning('If you experience error \"cudaSuccess out of memory\", try reducing number of threads you\'re using') Avalon.warning('If you experience error \"cudaSuccess out of memory\", try reducing number of threads you\'re using')
else: else:
try: with contextlib.suppress(ValueError):
# "0" is GPU ID. Both waifu2x drivers use the first GPU available, therefore only 0 makes sense # "0" is GPU ID. Both waifu2x drivers use the first GPU available, therefore only 0 makes sense
gpu_memory_available = (GPUtil.getGPUs()[0].memoryTotal - GPUtil.getGPUs()[0].memoryUsed) / 1024 gpu_memory_available = (GPUtil.getGPUs()[0].memoryTotal - GPUtil.getGPUs()[0].memoryUsed) / 1024
memory_status.append(('GPU', gpu_memory_available)) memory_status.append(('GPU', gpu_memory_available))
except ValueError:
pass
# go though each checkable memory type and check availability # go though each checkable memory type and check availability
for memory_type, memory_available in memory_status: for memory_type, memory_available in memory_status:
@ -225,7 +221,7 @@ def absolutify_paths(config):
# check waifu2x_ncnn_vulkan path # check waifu2x_ncnn_vulkan path
if not re.match('^[a-z]:', config['waifu2x_ncnn_vulkan']['waifu2x_ncnn_vulkan_path'], re.IGNORECASE): if not re.match('^[a-z]:', config['waifu2x_ncnn_vulkan']['waifu2x_ncnn_vulkan_path'], re.IGNORECASE):
config['waifu2x_ncnn_vulkan']['waifu2x_ncnn_vulkan_path'] = current_directory / config['waifu2x_ncnn_vulkan']['waifu2x_ncnn_vulkan_path'] config['waifu2x_ncnn_vulkan']['waifu2x_ncnn_vulkan_path'] = current_directory / config['waifu2x_ncnn_vulkan']['waifu2x_ncnn_vulkan_path']
# check anime4k path # check anime4k path
if not re.match('^[a-z]:', config['anime4k']['anime4k_path'], re.IGNORECASE): if not re.match('^[a-z]:', config['anime4k']['anime4k_path'], re.IGNORECASE):
config['anime4k']['anime4k_path'] = current_directory / config['anime4k']['anime4k_path'] config['anime4k']['anime4k_path'] = current_directory / config['anime4k']['anime4k_path']
@ -440,8 +436,6 @@ except Exception:
traceback.print_exc() traceback.print_exc()
finally: finally:
# remove Video2X cache directory # remove Video2X cache directory
try: with contextlib.suppress(FileNotFoundError):
if not preserve_frames: if not preserve_frames:
shutil.rmtree(video2x_cache_directory) shutil.rmtree(video2x_cache_directory)
except FileNotFoundError:
pass