@@ -23,7 +23,7 @@ def __call__(self, parser, namespace, values, option_string=None):
23
23
setattr (namespace , self .dest , os .path .abspath (os .path .expanduser (values )))
24
24
25
25
exit_code = 0
26
-
26
+
27
27
parser = argparse .ArgumentParser ()
28
28
subparsers = parser .add_subparsers ()
29
29
@@ -52,9 +52,9 @@ def process_extract(arguments):
52
52
p .add_argument ('--output-debug' , action = "store_true" , dest = "output_debug" , default = None , help = "Writes debug images to <output-dir>_debug\ directory." )
53
53
p .add_argument ('--no-output-debug' , action = "store_false" , dest = "output_debug" , default = None , help = "Don't writes debug images to <output-dir>_debug\ directory." )
54
54
p .add_argument ('--face-type' , dest = "face_type" , choices = ['half_face' , 'full_face' , 'whole_face' , 'head' , 'mark_only' ], default = None )
55
- p .add_argument ('--max-faces-from-image' , type = int , dest = "max_faces_from_image" , default = None , help = "Max faces from image." )
55
+ p .add_argument ('--max-faces-from-image' , type = int , dest = "max_faces_from_image" , default = None , help = "Max faces from image." )
56
56
p .add_argument ('--image-size' , type = int , dest = "image_size" , default = None , help = "Output image size." )
57
- p .add_argument ('--jpeg-quality' , type = int , dest = "jpeg_quality" , default = None , help = "Jpeg quality." )
57
+ p .add_argument ('--jpeg-quality' , type = int , dest = "jpeg_quality" , default = None , help = "Jpeg quality." )
58
58
p .add_argument ('--manual-fix' , action = "store_true" , dest = "manual_fix" , default = False , help = "Enables manual extract only frames where faces were not recognized." )
59
59
p .add_argument ('--manual-output-debug-fix' , action = "store_true" , dest = "manual_output_debug_fix" , default = False , help = "Performs manual reextract input-dir frames which were deleted from [output_dir]_debug\ dir." )
60
60
p .add_argument ('--manual-window-size' , type = int , dest = "manual_window_size" , default = 1368 , help = "Manual fix window size. Default: 1368." )
@@ -98,7 +98,7 @@ def process_util(arguments):
98
98
io .log_info ("Performing faceset unpacking...\r \n " )
99
99
from samplelib import PackedFaceset
100
100
PackedFaceset .unpack ( Path (arguments .input_dir ) )
101
-
101
+
102
102
if arguments .export_faceset_mask :
103
103
io .log_info ("Exporting faceset mask..\r \n " )
104
104
Util .export_faceset_mask ( Path (arguments .input_dir ) )
@@ -149,10 +149,10 @@ def process_train(arguments):
149
149
p .add_argument ('--cpu-only' , action = "store_true" , dest = "cpu_only" , default = False , help = "Train on CPU." )
150
150
p .add_argument ('--force-gpu-idxs' , dest = "force_gpu_idxs" , default = None , help = "Force to choose GPU indexes separated by comma." )
151
151
p .add_argument ('--silent-start' , action = "store_true" , dest = "silent_start" , default = False , help = "Silent start. Automatically chooses Best GPU and last used model." )
152
-
152
+
153
153
p .add_argument ('--execute-program' , dest = "execute_program" , default = [], action = 'append' , nargs = '+' )
154
154
p .set_defaults (func = process_train )
155
-
155
+
156
156
def process_exportdfm (arguments ):
157
157
osex .set_process_lowest_prio ()
158
158
from mainscripts import ExportDFM
@@ -266,8 +266,8 @@ def process_faceset_enhancer(arguments):
266
266
p .add_argument ('--force-gpu-idxs' , dest = "force_gpu_idxs" , default = None , help = "Force to choose GPU indexes separated by comma." )
267
267
268
268
p .set_defaults (func = process_faceset_enhancer )
269
-
270
-
269
+
270
+
271
271
p = facesettool_parser .add_parser ("resize" , help = "Resize DFL faceset." )
272
272
p .add_argument ('--input-dir' , required = True , action = fixPathAction , dest = "input_dir" , help = "Input directory of aligned faces." )
273
273
@@ -285,22 +285,22 @@ def process_dev_test(arguments):
285
285
p = subparsers .add_parser ( "dev_test" , help = "" )
286
286
p .add_argument ('--input-dir' , required = True , action = fixPathAction , dest = "input_dir" )
287
287
p .set_defaults (func = process_dev_test )
288
-
288
+
289
289
# ========== XSeg
290
290
xseg_parser = subparsers .add_parser ( "xseg" , help = "XSeg tools." ).add_subparsers ()
291
-
291
+
292
292
p = xseg_parser .add_parser ( "editor" , help = "XSeg editor." )
293
293
294
294
def process_xsegeditor (arguments ):
295
295
osex .set_process_lowest_prio ()
296
296
from XSegEditor import XSegEditor
297
297
global exit_code
298
298
exit_code = XSegEditor .start (Path (arguments .input_dir ))
299
-
299
+
300
300
p .add_argument ('--input-dir' , required = True , action = fixPathAction , dest = "input_dir" )
301
301
302
302
p .set_defaults (func = process_xsegeditor )
303
-
303
+
304
304
p = xseg_parser .add_parser ( "apply" , help = "Apply trained XSeg model to the extracted faces." )
305
305
306
306
def process_xsegapply (arguments ):
@@ -310,26 +310,26 @@ def process_xsegapply(arguments):
310
310
p .add_argument ('--input-dir' , required = True , action = fixPathAction , dest = "input_dir" )
311
311
p .add_argument ('--model-dir' , required = True , action = fixPathAction , dest = "model_dir" )
312
312
p .set_defaults (func = process_xsegapply )
313
-
314
-
313
+
314
+
315
315
p = xseg_parser .add_parser ( "remove" , help = "Remove applied XSeg masks from the extracted faces." )
316
316
def process_xsegremove (arguments ):
317
317
osex .set_process_lowest_prio ()
318
318
from mainscripts import XSegUtil
319
319
XSegUtil .remove_xseg (Path (arguments .input_dir ) )
320
320
p .add_argument ('--input-dir' , required = True , action = fixPathAction , dest = "input_dir" )
321
321
p .set_defaults (func = process_xsegremove )
322
-
323
-
322
+
323
+
324
324
p = xseg_parser .add_parser ( "remove_labels" , help = "Remove XSeg labels from the extracted faces." )
325
325
def process_xsegremovelabels (arguments ):
326
326
osex .set_process_lowest_prio ()
327
327
from mainscripts import XSegUtil
328
328
XSegUtil .remove_xseg_labels (Path (arguments .input_dir ) )
329
329
p .add_argument ('--input-dir' , required = True , action = fixPathAction , dest = "input_dir" )
330
330
p .set_defaults (func = process_xsegremovelabels )
331
-
332
-
331
+
332
+
333
333
p = xseg_parser .add_parser ( "fetch" , help = "Copies faces containing XSeg polygons in <input_dir>_xseg dir." )
334
334
335
335
def process_xsegfetch (arguments ):
@@ -338,7 +338,7 @@ def process_xsegfetch(arguments):
338
338
XSegUtil .fetch_xseg (Path (arguments .input_dir ) )
339
339
p .add_argument ('--input-dir' , required = True , action = fixPathAction , dest = "input_dir" )
340
340
p .set_defaults (func = process_xsegfetch )
341
-
341
+
342
342
def bad_args (arguments ):
343
343
parser .print_help ()
344
344
exit (0 )
@@ -349,9 +349,9 @@ def bad_args(arguments):
349
349
350
350
if exit_code == 0 :
351
351
print ("Done." )
352
-
352
+
353
353
exit (exit_code )
354
-
354
+
355
355
'''
356
356
import code
357
357
code.interact(local=dict(globals(), **locals()))
0 commit comments