model arg
This commit is contained in:
1 parent
7026da22f8
commit
8b291db30b
1 file changed
+11
+11
@@ -17,6 +17,7 @@ default_arg={
|
||||
'is_bongo': False,
|
||||
'cache_simplify': 1,
|
||||
'cache_size': 1,
|
||||
'model_type': 0,
|
||||
}
|
||||
|
||||
try:
|
||||
@@ -62,6 +63,12 @@ ttk.Label(launcher, text="iFacialMocap IP:Port").pack(fill='x', expand=True)
|
||||
ifm = tk.StringVar(value=args['ifm'])
|
||||
ttk.Entry(launcher, textvariable=ifm, state=False).pack(fill='x', expand=True)
|
||||
|
||||
ttk.Label(launcher, text="Model Simplify").pack(fill='x', expand=True)
|
||||
model_type = tk.IntVar(value=args['model_type'])
|
||||
ttk.Radiobutton(launcher, text='Off', value=0, variable=model_type).pack(fill='x', expand=True)
|
||||
ttk.Radiobutton(launcher, text='Low', value=1, variable=model_type).pack(fill='x', expand=True)
|
||||
ttk.Radiobutton(launcher, text='High', value=2, variable=model_type).pack(fill='x', expand=True)
|
||||
|
||||
ttk.Label(launcher, text="Facial Input Simplify").pack(fill='x', expand=True)
|
||||
cache_simplify = tk.IntVar(value=args['cache_simplify'])
|
||||
ttk.Radiobutton(launcher, text='Off', value=0, variable=cache_simplify).pack(fill='x', expand=True)
|
||||
@@ -115,6 +122,7 @@ def launch():
|
||||
'is_bongo': is_bongo.get(),
|
||||
'cache_simplify': cache_simplify.get(),
|
||||
'cache_size': cache_size.get(),
|
||||
'model_type': model_type.get(),
|
||||
}
|
||||
f = open('launcher.json', mode='w')
|
||||
json.dump(args, f)
|
||||
@@ -168,6 +176,9 @@ def launch():
|
||||
run_args.append(['0b','256mb','1gb','2gb','4gb','8gb'][args['cache_size']])
|
||||
run_args.append('--gpu_cache')
|
||||
run_args.append(['0b','128mb','512mb','1gb','2gb','4gb'][args['cache_size']])
|
||||
if args['model_type'] is not None:
|
||||
run_args.append('--model')
|
||||
run_args.append(['standard_float','standard_half','separable_half'][args['model_type']])
|
||||
run_args.append('--output_size')
|
||||
run_args.append('512x512')
|
||||
print('Launched: '+' '.join(run_args))
|
||||
|
||||
Reference in new issue
Block a user