david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Extractor: added warning messages if aligned folder contains files that will be deleted.

Browse Source
This commit is contained in:
iperov
2019-04-06 10:01:59 +04:00
parent cfae9f9fd8
commit 9f68b25987
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mainscripts/Extractor.py
+5 -2
View File
@@ -682,8 +682,11 @@ def main(input_dir,
if output_path.exists(): if output_path.exists():
if not manual_output_debug_fix and input_path != output_path: if not manual_output_debug_fix and input_path != output_path:
for filename in Path_utils.get_image_paths(output_path): output_images_paths = Path_utils.get_image_paths(output_path)
Path(filename).unlink() if len(output_images_paths) > 0:
io.input_bool("WARNING !!! \n %s contains files! \n They will be deleted. \n Press enter to continue." % (str(output_path)), False )
for filename in output_images_paths:
Path(filename).unlink()
else: else:
output_path.mkdir(parents=True, exist_ok=True) output_path.mkdir(parents=True, exist_ok=True)