github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-16 10:03:41 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

FacesetEnhancer now asks to merge aligned_enhanced/ to aligned/

Browse source
This commit is contained in:
Colombo 2019-12-27 11:09:15 +04:00
parent d46fb5cfd3
commit df8cf35c75
1 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
mainscripts/FacesetEnhancer.py
View file

@ -143,8 +143,8 @@ def process_folder ( dirpath, multi_gpu=False, cpu_only=False ):
dirpath_parts = '/'.join( dirpath.parts[-2:])
output_dirpath_parts = '/'.join( output_dirpath.parts[-2:] )
io.log_info (f"Enhancing faceset in {dirpath_parts}.")
io.log_info ( f"Processing to {output_dirpath_parts}.")
io.log_info (f"Enhancing faceset in {dirpath_parts}")
io.log_info ( f"Processing to {output_dirpath_parts}")
output_images_paths = Path_utils.get_image_paths(output_dirpath)
if len(output_images_paths) > 0:
@ -154,10 +154,12 @@ def process_folder ( dirpath, multi_gpu=False, cpu_only=False ):
image_paths = [Path(x) for x in Path_utils.get_image_paths( dirpath )]
result = FacesetEnhancerSubprocessor ( image_paths, output_dirpath, multi_gpu=multi_gpu, cpu_only=cpu_only).run()
io.log_info (f"Copying processed files to {dirpath_parts}.")
for (filepath, output_filepath) in result:
shutil.copy (output_filepath, filepath)
is_merge = io.input_bool (f"\r\nMerge {output_dirpath_parts} to {dirpath_parts} ? (y/n skip:y) : ", True)
if is_merge:
io.log_info (f"Copying processed files to {dirpath_parts}")
io.log_info (f"Removing {output_dirpath_parts}.")
shutil.rmtree(output_dirpath)
for (filepath, output_filepath) in result:
shutil.copy (output_filepath, filepath)
io.log_info (f"Removing {output_dirpath_parts}")
shutil.rmtree(output_dirpath)