def process_videoed_video_from_sequence(arguments):
osex.set_process_lowest_prio()
from mainscripts import VideoEd
VideoEd.video_from_sequence(arguments.input_dir, arguments.output_file,
arguments.reference_file, arguments.ext,
arguments.fps, arguments.bitrate,
arguments.lossless)
def process_train(arguments):
osex.set_process_lowest_prio()
kwargs = {
'model_class_name':
arguments.model_name,
'saved_models_path':
Path(arguments.model_dir),
'training_data_src_path':
Path(arguments.training_data_src_dir),
'training_data_dst_path':
Path(arguments.training_data_dst_dir),
'pretraining_data_path':
Path(arguments.pretraining_data_dir)
if arguments.pretraining_data_dir is not None else None,
'pretrained_model_path':
Path(arguments.pretrained_model_dir)
if arguments.pretrained_model_dir is not None else None,
'no_preview':
arguments.no_preview,
'force_model_name':
arguments.force_model_name,
'force_gpu_idxs':
[int(x) for x in arguments.force_gpu_idxs.split(',')]
if arguments.force_gpu_idxs is not None else None,
'cpu_only':
arguments.cpu_only,
'execute_programs':
[[int(x[0]), x[1]] for x in arguments.execute_program],
'debug':
arguments.debug,
}
from mainscripts import Trainer
Trainer.main(**kwargs)
def process_merge(arguments):
osex.set_process_lowest_prio()
kwargs = {
'model_class_name':
arguments.model_name,
'saved_models_path':
Path(arguments.model_dir),
'training_data_src_path':
Path(arguments.training_data_src_dir)
if arguments.training_data_src_dir is not None else None,
'force_model_name':
arguments.force_model_name,
'input_path':
Path(arguments.input_dir),
'output_path':
Path(arguments.output_dir),
'aligned_path':
Path(arguments.aligned_dir)
if arguments.aligned_dir is not None else None,
'cpu_only':
arguments.cpu_only,
'force_gpu_idxs':
arguments.force_gpu_idxs,
}
from mainscripts import Merger
Merger.main(**kwargs)
def process_faceset_enhancer(arguments):
osex.set_process_lowest_prio()
from mainscripts import FacesetEnhancer
FacesetEnhancer.process_folder ( Path(arguments.input_dir),
cpu_only=arguments.cpu_only,
force_gpu_idxs=arguments.force_gpu_idxs
)
def process_util(arguments):
osex.set_process_lowest_prio()
from mainscripts import Util
if arguments.add_landmarks_debug_images:
Util.add_landmarks_debug_images(input_path=arguments.input_dir)
if arguments.recover_original_aligned_filename:
Util.recover_original_aligned_filename(
input_path=arguments.input_dir)
if arguments.save_faceset_metadata:
Util.save_faceset_metadata_folder(input_path=arguments.input_dir)
if arguments.restore_faceset_metadata:
Util.restore_faceset_metadata_folder(
input_path=arguments.input_dir)
if arguments.pack_faceset:
io.log_info("Performing faceset packing...\r\n")
from samplelib import PackedFaceset
PackedFaceset.pack(Path(arguments.input_dir))
if arguments.unpack_faceset:
io.log_info("Performing faceset unpacking...\r\n")
from samplelib import PackedFaceset
PackedFaceset.unpack(Path(arguments.input_dir))
def process_videoed_cut_video(arguments):
osex.set_process_lowest_prio()
from mainscripts import VideoEd
VideoEd.cut_video (arguments.input_file,
arguments.from_time,
arguments.to_time,
arguments.audio_track_id,
arguments.bitrate)
def process_dev_extract_vggface2_dataset(arguments):
osex.set_process_lowest_prio()
from mainscripts import dev_misc
dev_misc.extract_vggface2_dataset( arguments.input_dir,
device_args={'cpu_only' : arguments.cpu_only,
'multi_gpu' : arguments.multi_gpu,
}
)
def process_dev_extract_umd_csv(arguments):
osex.set_process_lowest_prio()
from mainscripts import dev_misc
dev_misc.extract_umd_csv( arguments.input_csv_file,
device_args={'cpu_only' : arguments.cpu_only,
'multi_gpu' : arguments.multi_gpu,
}
)
def process_train(arguments):
osex.set_process_lowest_prio()
kwargs = {
'model_class_name':
arguments.model_name,
'saved_models_path':
Path(arguments.model_dir),
'training_data_src_path':
Path(arguments.training_data_src_dir),
'training_data_dst_path':
Path(arguments.training_data_dst_dir),
'pretraining_data_path':
Path(arguments.pretraining_data_dir)
if arguments.pretraining_data_dir is not None else None,
'pretrained_model_path':
Path(arguments.pretrained_model_dir)
if arguments.pretrained_model_dir is not None else None,
'no_preview':
arguments.no_preview,
'force_model_name':
arguments.force_model_name,
'force_gpu_idxs':
[int(x) for x in arguments.force_gpu_idxs.split(',')]
if arguments.force_gpu_idxs is not None else None,
'cpu_only':
arguments.cpu_only,
'silent_start':
arguments.silent_start,
'execute_programs':
[[int(x[0]), x[1]] for x in arguments.execute_program],
'debug':
arguments.debug,
'use_amp':
arguments.use_amp,
'opt':
arguments.opt,
'lr':
arguments.lr,
'decay_step':
arguments.decay_step,
'config_file':
arguments.config_file,
'bs_per_gpu':
arguments.bs_per_gpu,
}
if arguments.api == 'dfl':
from mainscripts import Trainer
Trainer.main(**kwargs)
elif arguments.api == 'tf1':
from mainscripts import Trainer_tf1
Trainer_tf1.main(**kwargs)
else:
print('Training API {} is invalid'.format(arguments.api))
exit(0)
def process_videoed_video_from_sequence(arguments):
osex.set_process_lowest_prio()
from mainscripts import VideoEd
VideoEd.video_from_sequence(input_dir=arguments.input_dir,
output_file=arguments.output_file,
reference_file=arguments.reference_file,
ext=arguments.ext,
fps=arguments.fps,
bitrate=arguments.bitrate,
include_audio=arguments.include_audio,
lossless=arguments.lossless)
def process_merge(arguments):
osex.set_process_lowest_prio()
from mainscripts import Merger
Merger.main ( model_class_name = arguments.model_name,
saved_models_path = Path(arguments.model_dir),
force_model_name = arguments.force_model_name,
input_path = Path(arguments.input_dir),
output_path = Path(arguments.output_dir),
output_mask_path = Path(arguments.output_mask_dir),
aligned_path = Path(arguments.aligned_dir) if arguments.aligned_dir is not None else None,
force_gpu_idxs = arguments.force_gpu_idxs,
cpu_only = arguments.cpu_only)
def process_extract(arguments):
osex.set_process_lowest_prio()
from mainscripts import Extractor
Extractor.main( detector = arguments.detector,
input_path = Path(arguments.input_dir),
output_path = Path(arguments.output_dir),
output_debug = arguments.output_debug,
manual_fix = arguments.manual_fix,
manual_output_debug_fix = arguments.manual_output_debug_fix,
manual_window_size = arguments.manual_window_size,
face_type = arguments.face_type,
cpu_only = arguments.cpu_only,
force_gpu_idxs = [ int(x) for x in arguments.force_gpu_idxs.split(',') ] if arguments.force_gpu_idxs is not None else None,
)
def process_dev_test(arguments):
osex.set_process_lowest_prio()
from mainscripts import dev_misc
dev_misc.dev_test(arguments.input_dir)
def process_videoed_denoise_image_sequence(arguments):
osex.set_process_lowest_prio()
from mainscripts import VideoEd
VideoEd.denoise_image_sequence(arguments.input_dir, arguments.factor)
def process_videoed_extract_video(arguments):
osex.set_process_lowest_prio()
from mainscripts import VideoEd
VideoEd.extract_video(arguments.input_file, arguments.output_dir,
arguments.output_ext, arguments.fps)
def process_exportdfm(arguments):
osex.set_process_lowest_prio()
from mainscripts import ExportDFM
ExportDFM.main(model_class_name=arguments.model_name,
saved_models_path=Path(arguments.model_dir))
def process_sort(arguments):
osex.set_process_lowest_prio()
from mainscripts import Sorter
Sorter.main(input_path=Path(arguments.input_dir),
sort_by_method=arguments.sort_by_method)
def process_faceset_resizer(arguments):
osex.set_process_lowest_prio()
from mainscripts import FacesetResizer
FacesetResizer.process_folder(Path(arguments.input_dir))
def process_xsegremovelabels(arguments):
osex.set_process_lowest_prio()
from mainscripts import XSegUtil
XSegUtil.remove_xseg_labels(Path(arguments.input_dir))
def process_xsegeditor(arguments):
osex.set_process_lowest_prio()
from XSegEditor import XSegEditor
global exit_code
exit_code = XSegEditor.start(Path(arguments.input_dir))
def process_dev_segmented_extract(arguments):
osex.set_process_lowest_prio()
from mainscripts import dev_misc
dev_misc.dev_segmented_extract(arguments.input_dir,
arguments.output_dir)
def process_xsegapply(arguments):
osex.set_process_lowest_prio()
from mainscripts import XSegUtil
XSegUtil.apply_xseg(Path(arguments.input_dir),
Path(arguments.model_dir))
def process_dev_apply_celebamaskhq(arguments):
osex.set_process_lowest_prio()
from mainscripts import dev_misc
dev_misc.apply_celebamaskhq( arguments.input_dir )
def process_xsegfetch(arguments):
osex.set_process_lowest_prio()
from mainscripts import XSegUtil
XSegUtil.fetch_xseg(Path(arguments.input_dir))
def process_xseg_split(arguments):
osex.set_process_lowest_prio()
from mainscripts import XSegUtil
XSegUtil.split(arguments.input_dir)
