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)