Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-03-29 18:47:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

added more exceptions for better error handling

Browse Source
This commit is contained in:
k4yt3x
2019-07-27 17:52:31 -04:00
parent 4639555626
commit 47e787c80c
3 changed files with 26 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
bin/video2x.py
View File

@@ -43,6 +43,7 @@ smooth and edges sharp.
"""
# local imports
from exceptions import *
from upscaler import Upscaler
# built-in imports
@@ -254,22 +255,22 @@ if args.version:
# arguments sanity check
if not args.input:
Avalon.error('You must specify input video file/directory path')
exit(1)
raise ArgumentError('input video path not specified')
if not args.output:
Avalon.error('You must specify output video file/directory path')
exit(1)
raise ArgumentError('output video path not specified')
if (args.driver == 'waifu2x_converter' or args.driver == 'waifu2x_ncnn_vulkan') and args.width and args.height:
Avalon.error('Waifu2x Converter CPP/NCNN accepts only scaling ratio')
exit(1)
raise ArgumentError('waifu2x-converter supports only scaling ratio')
if args.driver == 'waifu2x_ncnn_vulkan' and (args.ratio > 2 or not args.ratio.is_integer()):
Avalon.error('Scaling ratio must be 1 or 2 for waifu2x_ncnn_vulkan')
exit(1)
raise ArgumentError('scaling ratio must be 1 or 2 for waifu2x_ncnn_vulkan')
if (args.width or args.height) and args.ratio:
Avalon.error('You can only specify either scaling ratio or output width and height')
exit(1)
raise ArgumentError('both scaling ration and width/height specified')
if (args.width and not args.height) or (not args.width and args.height):
Avalon.error('You must specify both width and height')
exit(1)
raise ArgumentError('only one of width or height is specified')
# check available memory
check_memory()