43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
import argparse
|
|
import re
|
|
|
|
def convert_to_byte(size):
|
|
result = re.search('(\d+\.?\d*)(b|kb|mb|gb|tb)', size.lower())
|
|
if (result and result.groups()):
|
|
unit = result.groups()[1]
|
|
amount = float(result.groups()[0])
|
|
index = ['b', 'kb', 'mb', 'gb', 'tb'].index(unit)
|
|
return amount * pow(1024, index)
|
|
raise ValueError("Invalid size provided, value is " + size)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--debug', action='store_true')
|
|
parser.add_argument('--extend_movement', type=float)
|
|
parser.add_argument('--input', type=str, default='cam')
|
|
parser.add_argument('--character', type=str, default='y')
|
|
parser.add_argument('--output_dir', type=str)
|
|
parser.add_argument('--output_webcam', type=str)
|
|
parser.add_argument('--output_size', type=str, default='256x256')
|
|
parser.add_argument('--debug_input', action='store_true')
|
|
parser.add_argument('--mouse_input', type=str)
|
|
parser.add_argument('--perf', type=str)
|
|
parser.add_argument('--skip_model', action='store_true')
|
|
parser.add_argument('--ifm', type=str)
|
|
parser.add_argument('--anime4k', action='store_true')
|
|
parser.add_argument('--alpha_split', action='store_true')
|
|
parser.add_argument('--bongo', action='store_true')
|
|
parser.add_argument('--cache', type=str, default='256mb')
|
|
parser.add_argument('--gpu_cache', type=str, default='512mb')
|
|
parser.add_argument('--simplify', type=int, default=1)
|
|
args = parser.parse_args()
|
|
args.output_w = int(args.output_size.split('x')[0])
|
|
args.output_h = int(args.output_size.split('x')[1])
|
|
if args.cache is not None:
|
|
args.max_cache_len=int(convert_to_byte(args.cache)/262144/4)
|
|
else:
|
|
args.max_cache_len=0
|
|
if args.gpu_cache is not None:
|
|
args.max_gpu_cache_len=int(convert_to_byte(args.gpu_cache)/589824/4)
|
|
else:
|
|
args.max_gpu_cache_len=0
|
|
if args.output_webcam is None and args.output_dir is None: args.debug = True |